up

ecml_tools/create/__init__.py

@@ -33,7 +33,7 @@ def initialise(
             raise Exception(f"{path} already exists. Use --force to overwrite.")
 
         with cache_context(cache_dir):
-            obj = cls.from_config(partial=True, path=path, config=config, print=print)
+            obj = cls.from_config(path=path, config=config, print=print)
             obj.initialise(check_name=not no_check_name)
 
     @classmethod

ecml_tools/create/config.py

@@ -113,9 +113,6 @@ def normalise(self):
         # TODO: consider 2D grid points
         self.statistics_axis = statistics_axis
 
-    def input_handler(self, partial=False):
-        return InputHandler(self, partial=partial)
-
     @classmethod
     def _get_first_key_if_dict(cls, x):
         if isinstance(x, str):

ecml_tools/create/loaders.py

@@ -18,14 +18,16 @@
 from .config import loader_config
 from .utils import bytes, compute_directory_sizes, normalize_and_check_dates
 from .writer import DataWriter
+from .input import InputHandler
 
 LOG = logging.getLogger(__name__)
 
 VERSION = "0.13"
 
 
 class Creator:
-    def __init__(self, *, path, config, print=print, partial=False, **kwargs):
+    partial = False
+    def __init__(self, *, path, config, print=print, **kwargs):
         # Catch all floating point errors, including overflow, sqrt(<0), etc
         np.seterr(all="raise")
 
@@ -34,7 +36,8 @@ def __init__(self, *, path, config, print=print, partial=False, **kwargs):
         assert isinstance(path, str), path
 
         self.main_config = loader_config(config)
-        self.input_handler = self.main_config.input_handler(partial)
+        self.input_handler = InputHandler(self.main_config, partial=self.partial)
+
         self.path = path
         self.kwargs = kwargs
         self.print = print
@@ -102,6 +105,7 @@ def _add_dataset(self, mode="r+", **kwargs):
 
 
 class InitialiseCreator(Creator):
+    partial = True
     def initialise(self, check_name=True):
         """Create empty dataset"""