Skip to content

Commit b7acb89

Browse files
committed
use DataSourceBase to not test all strings
1 parent 9a14e55 commit b7acb89

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

intake/catalog/base.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
import re
1111
import time
1212

13-
from ..source.base import DataSource, NoEntry
13+
from ..source.base import DataSource, NoEntry, DataSourceBase
1414
from .utils import reload_on_change
1515

1616
logger = logging.getLogger('intake')
@@ -276,8 +276,8 @@ def serialize(self):
276276
kw = entry._captured_init_kwargs.copy()
277277
kw.pop('catalog', None)
278278
kw['parameters'] = {k.name: k.__getstate__()['kwargs'] for k in kw.get('parameters', [])}
279-
if not isinstance(kw['driver'], str):
280-
kw['driver'] = str(kw['driver']).split("'")[1]
279+
if issubclass(kw['driver'], DataSourceBase):
280+
kw['driver'] = ".".join([kw['driver'].__module__, kw['driver'].__name__])
281281
output["sources"][key] = kw
282282
return yaml.dump(output)
283283

0 commit comments

Comments
 (0)