power-grid-model-ds 1.1.4__py3-none-any.whl → 1.1.5__py3-none-any.whl

power_grid_model_ds/_core/data_source/generator/grid_generators.py

@@ -4,7 +4,7 @@
 
 """Generators for the grid"""
 
-from typing import Type
+from typing import Generic, Type, TypeVar
 
 import numpy as np
 
@@ -18,13 +18,15 @@ from power_grid_model_ds._core.model.grids.base import Grid
 
 # pylint: disable=too-few-public-methods,too-many-arguments,too-many-positional-arguments
 
+T = TypeVar("T", bound=Grid)
 
-class RadialGridGenerator:
+
+class RadialGridGenerator(Generic[T]):
     """Generates a random but structurally correct radial grid with the given specifications"""
 
     def __init__(
         self,
-        grid_class: Type[Grid],
+        grid_class: Type[T],
         nr_nodes: int = 100,
         nr_sources: int = 2,
         nr_nops: int = 10,
@@ -36,7 +38,7 @@ class RadialGridGenerator:
         self.nr_sources = nr_sources
         self.nr_nops = nr_nops
 
-    def run(self, seed=None, create_10_3_kv_net: bool = False):
+    def run(self, seed=None, create_10_3_kv_net: bool = False) -> T:
         """Run the generator to create a random radial grid.
 
         if a seed is provided, this will be used to set rng.

{power_grid_model_ds-1.1.4.dist-info → power_grid_model_ds-1.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: power-grid-model-ds
-Version: 1.1.4
+Version: 1.1.5
 Summary: Power Grid Model extension which provides a grid data structure for simulation and analysis
 Author-email: Contributors to the Power Grid Model project <powergridmodel@lfenergy.org>
 License: MPL-2.0

{power_grid_model_ds-1.1.4.dist-info → power_grid_model_ds-1.1.5.dist-info}/RECORD

@@ -11,7 +11,7 @@ power_grid_model_ds/_core/fancypy.py,sha256=MVnt6gRe2civTQBY17Dtp_wyqARpy4jP1NCp
 power_grid_model_ds/_core/load_flow.py,sha256=qpQkkcTV_-43sg0TwF44gahhWWbg2cqFHdVaecB0GQ0,4959
 power_grid_model_ds/_core/data_source/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 power_grid_model_ds/_core/data_source/generator/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-power_grid_model_ds/_core/data_source/generator/grid_generators.py,sha256=zToluoDgjn_KOC-hz4F6M377o3AMztcliP2QAWB_Q_E,2910
+power_grid_model_ds/_core/data_source/generator/grid_generators.py,sha256=knAtQOhdUru3z6Kzc0rjSE9PPeC4_c7SUVVLqTi3ZBU,2972
 power_grid_model_ds/_core/data_source/generator/arrays/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 power_grid_model_ds/_core/data_source/generator/arrays/base.py,sha256=Q0abhwyxOvZw-yn0m_vElSLL1mL3qdTNFzVWv-wCUd4,689
 power_grid_model_ds/_core/data_source/generator/arrays/line.py,sha256=Wh-tKHM5iOojGvJnql0atjuz5kjTnI5JWJzsG8p8ECg,6326
@@ -57,8 +57,8 @@ power_grid_model_ds/_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm
 power_grid_model_ds/_core/utils/misc.py,sha256=aAdlXjyKN6_T4b_CQZ879TbkbDZW4TwBUsxTpkwvymU,1228
 power_grid_model_ds/_core/utils/pickle.py,sha256=LGeTc7nu9RY1noOzLJzYaSHSWIgqzHy2xhmueKGuipc,1445
 power_grid_model_ds/_core/utils/zip.py,sha256=9RtJYhjlgNZOtxr4iI-CpsjT1axw5kCCqprfTjaIsiI,2197
-power_grid_model_ds-1.1.4.dist-info/LICENSE,sha256=GpbnG1pNl-ORtSP6dmHeiZvwy36UFli6MEZy1XlmBqo,14906
-power_grid_model_ds-1.1.4.dist-info/METADATA,sha256=XYSIcX5BJnpl0pG6N2vwSFRPJW5WO-QbXuw7MnBzc1E,8822
-power_grid_model_ds-1.1.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-power_grid_model_ds-1.1.4.dist-info/top_level.txt,sha256=nJa103Eqvm5TESYEKPFVImfLg_ugGOBznikrM-rZQZg,20
-power_grid_model_ds-1.1.4.dist-info/RECORD,,
+power_grid_model_ds-1.1.5.dist-info/LICENSE,sha256=GpbnG1pNl-ORtSP6dmHeiZvwy36UFli6MEZy1XlmBqo,14906
+power_grid_model_ds-1.1.5.dist-info/METADATA,sha256=61BFCXiDJ2zaf0LgHV13i0S4NlnVrw0AOD1pghs0FEI,8822
+power_grid_model_ds-1.1.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+power_grid_model_ds-1.1.5.dist-info/top_level.txt,sha256=nJa103Eqvm5TESYEKPFVImfLg_ugGOBznikrM-rZQZg,20
+power_grid_model_ds-1.1.5.dist-info/RECORD,,