power-grid-model-io 1.3.5__py3-none-any.whl → 1.3.7__py3-none-any.whl

power_grid_model_io/converters/__init__.py

@@ -8,3 +8,5 @@ Converters
 from power_grid_model_io.converters.pandapower_converter import PandaPowerConverter
 from power_grid_model_io.converters.pgm_json_converter import PgmJsonConverter
 from power_grid_model_io.converters.vision_excel_converter import VisionExcelConverter
+
+__all__ = ["PandaPowerConverter", "PgmJsonConverter", "VisionExcelConverter"]

power_grid_model_io/converters/base_converter.py

@@ -4,11 +4,13 @@
 """
 Abstract converter class
 """
+
 import logging
 from abc import ABC, abstractmethod
 from typing import Generic, Optional, Tuple, TypeVar
 
 import structlog
+from power_grid_model import DatasetType
 from power_grid_model.data_types import Dataset, SingleDataset
 
 from power_grid_model_io.data_stores.base_data_store import BaseDataStore
@@ -53,7 +55,9 @@ class BaseConverter(Generic[T], ABC):
 
         data = self._load_data(data)
         extra_info: ExtraInfo = {}
-        parsed_data = self._parse_data(data=data, data_type="input", extra_info=extra_info if make_extra_info else None)
+        parsed_data = self._parse_data(
+            data=data, data_type=DatasetType.input, extra_info=extra_info if make_extra_info else None
+        )
         if isinstance(parsed_data, list):
             raise TypeError("Input data can not be batch data")
         return parsed_data, extra_info
@@ -70,7 +74,7 @@ class BaseConverter(Generic[T], ABC):
 
         """
         data = self._load_data(data)
-        return self._parse_data(data=data, data_type="update", extra_info=None)
+        return self._parse_data(data=data, data_type=DatasetType.update, extra_info=None)
 
     def load_sym_output_data(self, data: Optional[T] = None) -> Dataset:
         """Load symmetric output data
@@ -84,7 +88,7 @@ class BaseConverter(Generic[T], ABC):
 
         """
         data = self._load_data(data)
-        return self._parse_data(data=data, data_type="sym_output", extra_info=None)
+        return self._parse_data(data=data, data_type=DatasetType.sym_output, extra_info=None)
 
     def load_asym_output_data(self, data: Optional[T] = None) -> Dataset:
         """Load asymmetric output data
@@ -98,7 +102,7 @@ class BaseConverter(Generic[T], ABC):
 
         """
         data = self._load_data(data)
-        return self._parse_data(data=data, data_type="asym_output", extra_info=None)
+        return self._parse_data(data=data, data_type=DatasetType.asym_output, extra_info=None)
 
     def load_sc_output_data(self, data: Optional[T] = None) -> Dataset:
         """Load sc output data
@@ -112,7 +116,7 @@ class BaseConverter(Generic[T], ABC):
 
         """
         data = self._load_data(data)
-        return self._parse_data(data=data, data_type="sc_output", extra_info=None)
+        return self._parse_data(data=data, data_type=DatasetType.sc_output, extra_info=None)
 
     def convert(self, data: Dataset, extra_info: Optional[ExtraInfo] = None) -> T:
         """Convert input/update/(a)sym_output data and optionally extra info.
@@ -181,7 +185,7 @@ class BaseConverter(Generic[T], ABC):
         raise ValueError("No data supplied!")
 
     @abstractmethod  # pragma: nocover
-    def _parse_data(self, data: T, data_type: str, extra_info: Optional[ExtraInfo]) -> Dataset:
+    def _parse_data(self, data: T, data_type: DatasetType, extra_info: Optional[ExtraInfo]) -> Dataset:
         pass
 
     @abstractmethod  # pragma: nocover