change default loader back to CSKG CSV pending #173

tetherless-world · Sep 10, 2020 · ad4d7ce · ad4d7ce
1 parent be46ef0
commit ad4d7ce
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mowgli_etl/_pipeline.py b/mowgli_etl/_pipeline.py
@@ -6,6 +6,7 @@
 from mowgli_etl._extractor import _Extractor
 from mowgli_etl._loader import _Loader
 from mowgli_etl._transformer import _Transformer
+from mowgli_etl.loader.cskg_csv.cskg_csv_loader import CskgCsvLoader
 from mowgli_etl.loader.kgtk.kgtk_edges_tsv_loader import KgtkEdgesTsvLoader
 
 
@@ -47,11 +48,13 @@ def __add_loader_arguments(cls, arg_parser):
 
     def __create_loader(self, id: str, loader: Optional[str] = None, **loader_kwds) -> _Loader:
         if loader is None:
-            loader = "kgtk_edges_tsv"
+            loader = "cskg_csv"
         else:
             loader = loader.lower()
 
-        if loader == "kgtk_edges_tsv":
+        if loader == "cskg_csv":
+            return CskgCsvLoader()
+        elif loader == "kgtk_edges_tsv":
             return KgtkEdgesTsvLoader()
         else:
             raise NotImplementedError(loader)