Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions src/power_grid_model_ds/_core/load_flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def calculate_power_flow(

Returns output of the power flow calculation (also stored in self.output_data)
"""
self.model = self.model or self._setup_model()
self.model = self.model or self.setup_model()

self.output_data = self.model.calculate_power_flow(
calculation_method=calculation_method, update_data=update_data, **kwargs
Expand Down Expand Up @@ -134,7 +134,7 @@ def update_model(self, update_data: Dict):


"""
self.model = self.model or self._setup_model()
self.model = self.model or self.setup_model()
self.model.update(update_data=update_data)

def update_grid(self) -> None:
Expand All @@ -154,7 +154,8 @@ def update_grid(self) -> None:
def _match_dtypes(first_dtype: np.dtype, second_dtype: np.dtype):
return list(set(first_dtype.names).intersection(set(second_dtype.names))) # type: ignore[arg-type]

def _setup_model(self):
def setup_model(self):
"""Setup the PowerGridModel with the input data."""
self.input_data = self.input_data or self.create_input_from_grid()
self.model = PowerGridModel(self.input_data, system_frequency=self.system_frequency)
return self.model
Loading