Merge pull request #58 from eminyous:add-scale-to-load

Add scale parameter to Loader.load method
eminyous · Feb 13, 2025 · f6707e7 · f6707e7
2 parents 0428454 + 5cb1b77
commit f6707e7
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/ocean/datasets/_load.py b/ocean/datasets/_load.py
@@ -19,16 +19,24 @@ def load(
         self,
         name: str,
         *,
+        scale: bool = False,
         return_mapper: Literal[True] = True,
     ) -> Loaded: ...
 
     @overload
-    def load(self, name: str, *, return_mapper: Literal[False]) -> Dataset: ...
+    def load(
+        self,
+        name: str,
+        *,
+        scale: bool = False,
+        return_mapper: Literal[False],
+    ) -> Dataset: ...
 
     def load(
         self,
         name: str,
         *,
+        scale: bool = False,
         return_mapper: bool = True,
     ) -> Dataset | Loaded:
         path = f"{name}/{name}.csv"
@@ -45,7 +53,7 @@ def load(
             data,
             discretes=discretes,
             encoded=encoded,
-            scale=False,
+            scale=scale,
         )
 
         if return_mapper: