File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 1010import re
1111import time
1212
13- from ..source .base import DataSource , NoEntry
13+ from ..source .base import DataSource , NoEntry , DataSourceBase
1414from .utils import reload_on_change
1515
1616logger = 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
You can’t perform that action at this time.
0 commit comments