LightSim2Grid 0.8.1.dev0__tar.gz → 0.8.2__tar.gz

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2/LightSim2Grid.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LightSim2Grid
-Version: 0.8.1.dev0
+Version: 0.8.2
 Summary: LightSim2Grid implements a c++ backend targeting the Grid2Op platform.
 Home-page: https://github.com/BDonnot/lightsim2grid/
 Author: Benjamin Donnot
@@ -50,7 +50,7 @@ Requires-Dist: numba; extra == "recommended"
 Provides-Extra: test
 Requires-Dist: grid2op>=1.6.4; extra == "test"
 Requires-Dist: numba; extra == "test"
-Requires-Dist: pandapower>=2.8.0; extra == "test"
+Requires-Dist: pandapower>=2.14.0; extra == "test"
 Requires-Dist: packaging; extra == "test"
 Requires-Dist: pypowsybl; extra == "test"
 
@@ -130,12 +130,13 @@ Since version 0.5.3, lightsim2grid is can be installed like most python packages
 `python -m pip install lightsim2grid`
 
 It includes faster grid2op backend and the `SuiteSparse` faster `KLU` solver, even on windows. This is definitely the 
-easiest method to install lightsim2grid on your system and have it running without too much issues.
+easiest method to install lightsim2grid on your system and have it running.
 
 Note though that these packages have been compiled on a different platform that the one you are using. You might still
-get some benefit (in terms of performances) to install it from your on your machine.
+get some benefit (in terms of performances) to install it from your on your machine with the proper compilations flags (
+see section [6.1 Customization of the compilation](#Customization-of-the-compilation) for more information)
 
-Pypi packages are available for linux, windows and macos with python versions: 
+Pypi packages are available for linux (`x86_64` cpu architecture), windows (`x86_64` cpu architecture) and macos (`x86_64` cpu architecture) with python versions: 
 
 - 3.7
 - 3.8
@@ -144,8 +145,17 @@ Pypi packages are available for linux, windows and macos with python versions:
 - 3.11 (lightsim2grid >= 0.7.1)
 - 3.12 (lightsim2grid >= 0.7.5)
 
+
+As from version 0.8.2, we also distribute windows `arm64` and macos `arm64` binaries of lightsim2grid that can be installed
+directly with pip too (requires python >= 3.8 for macos and python >= 3.9 for windows). We do not currently produce `arm64` (`aarch64`) linux binaries because it takes too long to build. If you really want them, let us know and we'll see what we can do.
+
+
 **NB** on some version of MacOs (thanks Apple !), especially the one using M1 or M2 chip, lightsim2grid is only available
-on pypi starting from version 0.7.3
+on pypi starting from version 0.7.3 We attempted to deliver arm64 lightsim2grid version but we could not test them. So if you 
+want a reliable working and tested version of lightsim2grid on newest version of macos (with M1 or M2 chips for example) please
+use lightsim2grid >= 0.8.2
+
+**NB** we do not currently build any 32 bits lightsim2grid libraries.
 
 ## Installation (from source, for more advanced user)
 
@@ -278,7 +288,7 @@ For example: `export PATH_NICSLU=/home/user/Documents/cktso`
 
 #### Enable 03 optimization
 By default, at least on ubuntu, only the "-O2" compiler flags is used. To use the O3 optimization flag, you need
-to specify the `__COMPLILE_O3` environment variable: `set __COMPLILE_O3=1` before the compilation (so before
+to specify the `__O3_OPTIM` environment variable: `set __O3_OPTIM=1` (or `$Env:__O3_OPTIM=1` in powershell) before the compilation (so before
 `python3 setup.py build` or `python -m pip install -e .`)
 
 This compilation argument will increase the compilation time, but will make the package faster.

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/LightSim2Grid.egg-info/SOURCES.txt

@@ -483,6 +483,7 @@ lightsim2grid/physical_law_checker.py
 lightsim2grid/securityAnalysis.py
 lightsim2grid/timeSerie.py
 lightsim2grid/_utils/__init__.py
+lightsim2grid/compilation_options/__init__.py
 lightsim2grid/elements/__init__.py
 lightsim2grid/gridmodel/__init__.py
 lightsim2grid/gridmodel/_aux_add_dc_line.py
@@ -500,6 +501,8 @@ lightsim2grid/gridmodel/from_pypowsybl.py
 lightsim2grid/gridmodel/initGridModel.py
 lightsim2grid/newtonpf/__init__.py
 lightsim2grid/newtonpf/newtonpf.py
+lightsim2grid/rewards/__init__.py
+lightsim2grid/rewards/n1ContingencyReward.py
 lightsim2grid/solver/__init__.py
 src/BaseConstants.cpp
 src/BaseConstants.h

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/LightSim2Grid.egg-info/requires.txt

@@ -25,6 +25,6 @@ numba
 [test]
 grid2op>=1.6.4
 numba
-pandapower>=2.8.0
+pandapower>=2.14.0
 packaging
 pypowsybl

{LightSim2Grid-0.8.1.dev0/LightSim2Grid.egg-info → lightsim2grid-0.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: LightSim2Grid
-Version: 0.8.1.dev0
+Version: 0.8.2
 Summary: LightSim2Grid implements a c++ backend targeting the Grid2Op platform.
 Home-page: https://github.com/BDonnot/lightsim2grid/
 Author: Benjamin Donnot
@@ -50,7 +50,7 @@ Requires-Dist: numba; extra == "recommended"
 Provides-Extra: test
 Requires-Dist: grid2op>=1.6.4; extra == "test"
 Requires-Dist: numba; extra == "test"
-Requires-Dist: pandapower>=2.8.0; extra == "test"
+Requires-Dist: pandapower>=2.14.0; extra == "test"
 Requires-Dist: packaging; extra == "test"
 Requires-Dist: pypowsybl; extra == "test"
 
@@ -130,12 +130,13 @@ Since version 0.5.3, lightsim2grid is can be installed like most python packages
 `python -m pip install lightsim2grid`
 
 It includes faster grid2op backend and the `SuiteSparse` faster `KLU` solver, even on windows. This is definitely the 
-easiest method to install lightsim2grid on your system and have it running without too much issues.
+easiest method to install lightsim2grid on your system and have it running.
 
 Note though that these packages have been compiled on a different platform that the one you are using. You might still
-get some benefit (in terms of performances) to install it from your on your machine.
+get some benefit (in terms of performances) to install it from your on your machine with the proper compilations flags (
+see section [6.1 Customization of the compilation](#Customization-of-the-compilation) for more information)
 
-Pypi packages are available for linux, windows and macos with python versions: 
+Pypi packages are available for linux (`x86_64` cpu architecture), windows (`x86_64` cpu architecture) and macos (`x86_64` cpu architecture) with python versions: 
 
 - 3.7
 - 3.8
@@ -144,8 +145,17 @@ Pypi packages are available for linux, windows and macos with python versions:
 - 3.11 (lightsim2grid >= 0.7.1)
 - 3.12 (lightsim2grid >= 0.7.5)
 
+
+As from version 0.8.2, we also distribute windows `arm64` and macos `arm64` binaries of lightsim2grid that can be installed
+directly with pip too (requires python >= 3.8 for macos and python >= 3.9 for windows). We do not currently produce `arm64` (`aarch64`) linux binaries because it takes too long to build. If you really want them, let us know and we'll see what we can do.
+
+
 **NB** on some version of MacOs (thanks Apple !), especially the one using M1 or M2 chip, lightsim2grid is only available
-on pypi starting from version 0.7.3
+on pypi starting from version 0.7.3 We attempted to deliver arm64 lightsim2grid version but we could not test them. So if you 
+want a reliable working and tested version of lightsim2grid on newest version of macos (with M1 or M2 chips for example) please
+use lightsim2grid >= 0.8.2
+
+**NB** we do not currently build any 32 bits lightsim2grid libraries.
 
 ## Installation (from source, for more advanced user)
 
@@ -278,7 +288,7 @@ For example: `export PATH_NICSLU=/home/user/Documents/cktso`
 
 #### Enable 03 optimization
 By default, at least on ubuntu, only the "-O2" compiler flags is used. To use the O3 optimization flag, you need
-to specify the `__COMPLILE_O3` environment variable: `set __COMPLILE_O3=1` before the compilation (so before
+to specify the `__O3_OPTIM` environment variable: `set __O3_OPTIM=1` (or `$Env:__O3_OPTIM=1` in powershell) before the compilation (so before
 `python3 setup.py build` or `python -m pip install -e .`)
 
 This compilation argument will increase the compilation time, but will make the package faster.

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/README.md

@@ -74,12 +74,13 @@ Since version 0.5.3, lightsim2grid is can be installed like most python packages
 `python -m pip install lightsim2grid`
 
 It includes faster grid2op backend and the `SuiteSparse` faster `KLU` solver, even on windows. This is definitely the 
-easiest method to install lightsim2grid on your system and have it running without too much issues.
+easiest method to install lightsim2grid on your system and have it running.
 
 Note though that these packages have been compiled on a different platform that the one you are using. You might still
-get some benefit (in terms of performances) to install it from your on your machine.
+get some benefit (in terms of performances) to install it from your on your machine with the proper compilations flags (
+see section [6.1 Customization of the compilation](#Customization-of-the-compilation) for more information)
 
-Pypi packages are available for linux, windows and macos with python versions: 
+Pypi packages are available for linux (`x86_64` cpu architecture), windows (`x86_64` cpu architecture) and macos (`x86_64` cpu architecture) with python versions: 
 
 - 3.7
 - 3.8
@@ -88,8 +89,17 @@ Pypi packages are available for linux, windows and macos with python versions:
 - 3.11 (lightsim2grid >= 0.7.1)
 - 3.12 (lightsim2grid >= 0.7.5)
 
+
+As from version 0.8.2, we also distribute windows `arm64` and macos `arm64` binaries of lightsim2grid that can be installed
+directly with pip too (requires python >= 3.8 for macos and python >= 3.9 for windows). We do not currently produce `arm64` (`aarch64`) linux binaries because it takes too long to build. If you really want them, let us know and we'll see what we can do.
+
+
 **NB** on some version of MacOs (thanks Apple !), especially the one using M1 or M2 chip, lightsim2grid is only available
-on pypi starting from version 0.7.3
+on pypi starting from version 0.7.3 We attempted to deliver arm64 lightsim2grid version but we could not test them. So if you 
+want a reliable working and tested version of lightsim2grid on newest version of macos (with M1 or M2 chips for example) please
+use lightsim2grid >= 0.8.2
+
+**NB** we do not currently build any 32 bits lightsim2grid libraries.
 
 ## Installation (from source, for more advanced user)
 
@@ -222,7 +232,7 @@ For example: `export PATH_NICSLU=/home/user/Documents/cktso`
 
 #### Enable 03 optimization
 By default, at least on ubuntu, only the "-O2" compiler flags is used. To use the O3 optimization flag, you need
-to specify the `__COMPLILE_O3` environment variable: `set __COMPLILE_O3=1` before the compilation (so before
+to specify the `__O3_OPTIM` environment variable: `set __O3_OPTIM=1` (or `$Env:__O3_OPTIM=1` in powershell) before the compilation (so before
 `python3 setup.py build` or `python -m pip install -e .`)
 
 This compilation argument will increase the compilation time, but will make the package faster.

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/lightsim2grid/__init__.py

@@ -1,13 +1,14 @@
-# Copyright (c) 2020, RTE (https://www.rte-france.com)
+# Copyright (c) 2020-2024, RTE (https://www.rte-france.com)
 # See AUTHORS.txt
 # This Source Code Form is subject to the terms of the Mozilla Public License, version 2.0.
 # If a copy of the Mozilla Public License, version 2.0 was not distributed with this file,
 # you can obtain one at http://mozilla.org/MPL/2.0/.
 # SPDX-License-Identifier: MPL-2.0
 # This file is part of LightSim2grid, LightSim2grid implements a c++ backend targeting the Grid2Op platform.
-__version__ = "0.8.1.dev0"
 
-__all__ = ["newtonpf", "SolverType", "ErrorType", "solver"]
+__version__ = "0.8.2"
+
+__all__ = ["newtonpf", "SolverType", "ErrorType", "solver", "compilation_options"]
 
 # import directly from c++ module
 from lightsim2grid.solver import SolverType
@@ -19,7 +20,7 @@ try:
 except ImportError as exc_:
     # grid2op is not installed, the Backend will not be available
     pass
-    print(f"LightSimBackend import error: {exc_}")
+    # print(f"LightSimBackend import error: {exc_}")
 
 try:
     from lightsim2grid.physical_law_checker import PhysicalLawChecker
@@ -27,7 +28,7 @@ try:
 except ImportError as exc_:
     # grid2op is not installed, the PhysicalLawChecker will not be available
     pass
-    print(f"PhysicalLawChecker import error: {exc_}")
+    # print(f"PhysicalLawChecker import error: {exc_}")
 
 try:
     from lightsim2grid.timeSerie import TimeSerie
@@ -36,7 +37,7 @@ try:
 except ImportError as exc_:
     # grid2op is not installed, the TimeSeries module will not be available
     pass
-    print(f"TimeSerie import error: {exc_}")
+    # print(f"TimeSerie import error: {exc_}")
 
 try:
     from lightsim2grid.contingencyAnalysis import ContingencyAnalysis
@@ -45,4 +46,12 @@ try:
 except ImportError as exc_:
     # grid2op is not installed, the SecurtiyAnalysis module will not be available
     pass
-    print(f"ContingencyAnalysis import error: {exc_}")
+    # print(f"ContingencyAnalysis import error: {exc_}")
+    
+try:
+    from lightsim2grid.rewards import *
+    __all__.append("rewards")
+except ImportError as exc_:
+    # grid2op is not installed, the SecurtiyAnalysis module will not be available
+    pass
+    # print(f"rewards import error: {exc_}")

lightsim2grid-0.8.2/lightsim2grid/compilation_options/__init__.py

@@ -0,0 +1,37 @@
+# Copyright (c) 2024, RTE (https://www.rte-france.com)
+# See AUTHORS.txt
+# This Source Code Form is subject to the terms of the Mozilla Public License, version 2.0.
+# If a copy of the Mozilla Public License, version 2.0 was not distributed with this file,
+# you can obtain one at http://mozilla.org/MPL/2.0/.
+# SPDX-License-Identifier: MPL-2.0
+# This file is part of LightSim2grid, LightSim2grid implements a c++ backend targeting the Grid2Op platform.
+
+__all__ = ["klu_solver_available",
+           "nicslu_solver_available",
+           "cktso_solver_available",
+           "compiled_march_native",
+           "compiled_o3_optim",
+           "version",
+           "lightsim2grid_lib"]
+
+from lightsim2grid_cpp import klu_solver_available
+from lightsim2grid_cpp import nicslu_solver_available
+from lightsim2grid_cpp import cktso_solver_available
+from lightsim2grid_cpp import compiled_march_native
+from lightsim2grid_cpp import compiled_o3_optim
+from lightsim2grid_cpp import version
+from lightsim2grid_cpp import __file__ as lightsim2grid_lib
+
+try:
+    from lightsim2grid_cpp import nicslu_lib
+    __all__.append("nicslu_lib")
+except ImportError :
+    # NICSLU linear solver is not available
+    pass
+
+try:
+    from lightsim2grid_cpp import cktso_lib
+    __all__.append("cktso_lib")
+except ImportError :
+    # CKTSO linear solver is not available
+    pass

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/lightsim2grid/contingencyAnalysis.py

@@ -6,21 +6,25 @@
 # SPDX-License-Identifier: MPL-2.0
 # This file is part of LightSim2grid, LightSim2grid implements a c++ backend targeting the Grid2Op platform.
 
-__all__ = ["ContingencyAnalysisCPP", "ContingencyAnalysis",
-           # deprecated
-           "SecurityAnalysisCPP", "SecurityAnalysis",
-           ]
+__all__ = ["ContingencyAnalysisCPP"]
 
 import copy
 import numpy as np
 from collections.abc import Iterable
 
-from lightsim2grid.lightSimBackend import LightSimBackend
 from lightsim2grid.solver import SolverType
 from lightsim2grid_cpp import ContingencyAnalysisCPP
 
+try:
+    from lightsim2grid.lightSimBackend import LightSimBackend
+    __all__.append("ContingencyAnalysis")
+    GRID2OP_INSTALLED = True
+except ImportError as exc_:
+    # grid2op is not installed
+    GRID2OP_INSTALLED = False
 
-class ContingencyAnalysis(object):
+
+class __ContingencyAnalysis(object):
     """
     This class allows to perform a "security analysis" from a given grid state.
 
@@ -76,10 +80,26 @@ class ContingencyAnalysis(object):
     STR_TYPES = (str, np.str_)  # np.str deprecated in numpy 1.20 and earlier versions not supported anyway
         
     def __init__(self, grid2op_env):
-        if not isinstance(grid2op_env.backend, LightSimBackend):
-            raise RuntimeError("This class only works with LightSimBackend")
-        self.grid2op_env = grid2op_env.copy()
-        self.computer = ContingencyAnalysisCPP(self.grid2op_env.backend._grid)
+        if not GRID2OP_INSTALLED:
+            raise RuntimeError("Impossible to use the python wrapper `ContingencyAnalysis` "
+                               "when grid2op is not installed. Please fall back to the "
+                               "c++ version (available in python) with:\n"
+                               "\tfrom lightsim2grid.contingencyAnalysis import ContingencyAnalysisCPP\n"
+                               "and refer to the appropriate documentation.")
+        from grid2op.Environment import Environment
+        if isinstance(grid2op_env, Environment):    
+            if not isinstance(grid2op_env.backend, LightSimBackend):
+                raise RuntimeError("This class only works with LightSimBackend")
+            self._ls_backend = grid2op_env.backend.copy()
+        elif isinstance(grid2op_env, LightSimBackend):
+            if hasattr(grid2op_env, "_is_loaded") and not grid2op_env._is_loaded:
+                raise RuntimeError("Impossible to init a `ContingencyAnalysis` "
+                                   "with a backend that has not been initialized.")
+            self._ls_backend = grid2op_env.copy()
+        else:
+            raise RuntimeError("`ContingencyAnalysis` can only be created "
+                               "with a grid2op `Environment` or a `LightSimBackend`")
+        self.computer = ContingencyAnalysisCPP(self._ls_backend._grid)
         self._contingency_order = {}  # key: contingency (as tuple), value: order in which it is entered
         self._all_contingencies = []
         self.__computed = False
@@ -100,16 +120,29 @@ class ContingencyAnalysis(object):
         raise RuntimeError("Impossible to add new topologies like this. Please use `add_single_contingency` "
                            "or `add_multiple_contingencies`.")
 
-    def clear(self):
+    # TODO implement that !
+    def __update_grid(self, backend_act):
+        self.clear(with_contlist=False)
+        self._ls_backend.apply_action(backend_act)
+        # run the powerflow
+        self._ls_backend.runpf()
+        # update the computer
+        # self.computer = ContingencyAnalysisCPP(self._ls_backend._grid)
+        # self.computer.update_grid(...)  # not implemented
+        
+    def clear(self, with_contlist=True):
         """
         Clear the list of contingencies to simulate
         """
-        self.computer.clear()
-        self._contingency_order = {}
         self.__computed = False
         self._vs = None
         self._ampss = None
-        self._all_contingencies = []
+        if with_contlist:
+            self.computer.clear()
+            self._contingency_order = {}
+            self._all_contingencies = []
+        else:
+            self.computer.clear_results_only()
 
     def _single_cont_to_li_int(self, single_cont):
         li_disc = []
@@ -118,7 +151,7 @@ class ContingencyAnalysis(object):
 
         for stuff in single_cont:
             if isinstance(stuff, type(self).STR_TYPES):
-                stuff = np.where(self.grid2op_env.name_line == stuff)
+                stuff = np.where(type(self._ls_backend).name_line == stuff)
                 stuff = stuff[0]
                 if stuff.size == 0:
                     # name is not found
@@ -233,7 +266,7 @@ class ContingencyAnalysis(object):
             for single_cont_id in range(env.n_line):
                 self.add_single_contingency(single_cont_id)
         """
-        for single_cont_id in range(self.grid2op_env.n_line):
+        for single_cont_id in range(type(self._ls_backend).n_line):
             self.add_single_contingency(single_cont_id)
 
     def get_flows(self, *args):
@@ -302,10 +335,10 @@ class ContingencyAnalysis(object):
             been entered. Please use `get_flows()` method for easier reading back of the results
 
         """
-        v_init = self.grid2op_env.backend.V
+        v_init = 1. * self._ls_backend.V
         self.computer.compute(v_init,
-                              self.grid2op_env.backend.max_it,
-                              self.grid2op_env.backend.tol)
+                              self._ls_backend.max_it,
+                              self._ls_backend.tol)
         self._vs = self.computer.get_voltages()
         self.__computed = True
         return self._vs
@@ -342,6 +375,10 @@ class ContingencyAnalysis(object):
 
     def close(self):
         """permanently close the object"""
-        self.grid2op_env.close()
+        self._ls_backend.close()
         self.clear()
         self.computer.close()
+        
+        
+if GRID2OP_INSTALLED:
+    ContingencyAnalysis = __ContingencyAnalysis

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/lightsim2grid/gridmodel/from_pypowsybl.py

@@ -10,6 +10,7 @@ import warnings
 import numpy as np
 import pandas as pd
 import pypowsybl as pypo
+from typing import Union
 
 from lightsim2grid_cpp import GridModel
 
@@ -36,7 +37,7 @@ def _aux_get_bus(bus_df, df, conn_key="connected", bus_key="bus_id"):
 
 
 def init(net : pypo.network,
-         gen_slack_id: int = None,
+         gen_slack_id: Union[int, str] = None,
          slack_bus_id: int = None,
          sn_mva = 100.,
          sort_index=True,
@@ -78,8 +79,8 @@ def init(net : pypo.network,
     # to handle encoding in 32 bits and overflow when "splitting" the Q values among 
     min_q = df_gen["min_q"].values.astype(np.float32)
     max_q = df_gen["max_q"].values.astype(np.float32)
-    min_q[~np.isfinite(min_q)] = np.finfo(np.float32).min * 0.5 + 1.
-    max_q[~np.isfinite(max_q)] = np.finfo(np.float32).max * 0.5 - 1.
+    min_q[~np.isfinite(min_q)] = np.finfo(np.float32).min * 1e-4 + 1.
+    max_q[~np.isfinite(max_q)] = np.finfo(np.float32).max * 1e-4 - 1.
     gen_bus, gen_disco = _aux_get_bus(bus_df, df_gen)
     model.init_generators_full(df_gen["target_p"].values,
                                df_gen["target_v"].values / voltage_levels.loc[df_gen["voltage_level_id"].values]["nominal_v"].values,
@@ -260,7 +261,17 @@ def init(net : pypo.network,
     elif gen_slack_id is not None:
         if slack_bus_id is not None:
             raise RuntimeError(f"You provided both gen_slack_id and slack_bus_id which is not possible.")
-        model.add_gen_slackbus(gen_slack_id, 1.)
+        
+        if isinstance(gen_slack_id, str):
+            gen_slack_id_int = int((df_gen.index == gen_slack_id).nonzero()[0][0])
+        else:
+            try:
+                gen_slack_id_int = int(gen_slack_id)
+            except Exception:
+                raise RuntimeError("'slack_bus_id' should be either an int or a generator names")
+            if gen_slack_id_int != gen_slack_id:
+                raise RuntimeError("'slack_bus_id' should be either an int or a generator names")
+        model.add_gen_slackbus(gen_slack_id_int, 1.)
     elif slack_bus_id is not None:
         gen_bus = np.array([el.bus_id for el in model.get_generators()])
         gen_is_conn_slack = gen_bus == model._orig_to_ls[slack_bus_id]

{LightSim2Grid-0.8.1.dev0 → lightsim2grid-0.8.2}/lightsim2grid/gridmodel/initGridModel.py

@@ -15,6 +15,9 @@ __all__ = ["init", "GridModel"]
 import numpy as np
 from numbers import Number
 import warnings
+
+import pandapower
+import lightsim2grid
 from lightsim2grid_cpp import GridModel, PandaPowerConverter
 from lightsim2grid.gridmodel._aux_add_sgen import _aux_add_sgen
 from lightsim2grid.gridmodel._aux_add_load import _aux_add_load
@@ -28,7 +31,7 @@ from lightsim2grid.gridmodel._aux_add_storage import _aux_add_storage
 from lightsim2grid.gridmodel._aux_add_dc_line import _aux_add_dc_line
 
 
-def init(pp_net):
+def init(pp_net: "pandapower.auxiliary.pandapowerNet") -> GridModel:
     """
     Convert a pandapower network as input into a GridModel.
 
@@ -53,7 +56,7 @@ def init(pp_net):
 
     Parameters
     ----------
-    pp_net: :class:`pandapower.grid`
+    pp_net: :class:`pandapower.auxiliary.pandapowerNet`
         The initial pandapower network you want to convert
 
     Returns