domainiac 9.3.1__py3-none-any.whl → 10.0.0__py3-none-any.whl

domainiac/__init__.py

@@ -1,9 +1,8 @@
 from .managers import (
     AvailabilityManager,
+    MasterdataManager,
     MeteringManager,
     NWPManager,
-    PlantManager,
     ResourceManager,
-    UnitManager,
 )
 from .modeling import Coordinate, Group, Neighborhood, NWPParameter, NWPProvider, Plant

domainiac/managers/__init__.py

@@ -3,6 +3,4 @@ from .masterdata_manager import MasterdataManager
 from .metering_manager import MeteringManager
 from .nwp_manager import NWPManager
 from .outage_manager import OutageManager
-from .plant_manager import PlantManager
 from .resource_manager import ResourceManager
-from .unit_manager import UnitManager

domainiac/managers/masterdata_manager.py

@@ -1,79 +1,229 @@
+import json
+from pathlib import Path
+
 import datamazing.pandas as pdz
 import pandas as pd
-from typeguard import typechecked
+import utm
 
-from ..wrappers import cache_decorator
+CONNECTION_POINT_BEHIND_THE_METER = "installationstilsluttet (I)"
 
 
 class MasterdataManager:
-    """
-    Manager which simplifies the process of getting units from masterdata.
-    """
+
+    SCHEMA = json.loads(
+        (Path(__file__).parent / "schemas/masterdata_manager.json").read_bytes()
+    )
 
     def __init__(
         self,
         db: pdz.Database,
-        time_interval: pdz.TimeInterval,
-        resolution: pd.Timedelta,
-        cache_masterdata: bool = False,
+        time_interval: pdz.TimeInterval = None,
+        as_of_time: pd.Timestamp = None,
     ) -> None:
         self.db = db
         self.time_interval = time_interval
-        self.resolution = resolution
-        self.cache_masterdata = cache_masterdata
+        self.as_of_time = as_of_time
 
-    masterdata_cache = {}
+        if as_of_time is not None and as_of_time.utcoffset().total_seconds() != 0:
+            raise ValueError("'as_of_time' must be utc")
 
-    @typechecked
-    def _get_operational_entities(self, table: str) -> pd.DataFrame:
-        filters = {"standing_entity_state": "InOperation"}
-        df = self.db.query(table, filters=filters)
-        df = df[df["decommission_date_utc"].isna()].reset_index(drop=True)
-        return df
+        if as_of_time is not None and (time_interval is not None):
+            raise ValueError("Cannot provide both 'as_of_time' and 'time_interval'")
+        if as_of_time is None and (time_interval is None):
+            raise ValueError(
+                "Either 'as_of_time' must be provided, or 'time_interval' must be"
+                "provided"
+            )
 
-    @typechecked
-    def get_operational_entities(self, table: str) -> pd.DataFrame:
-        """Gets the operational data for a given table."""
+    @property
+    def start_time(self) -> pd.Timestamp:
+        if self.as_of_time:
+            return self.as_of_time
+        else:
+            return self.time_interval.left
 
-        if self.cache_masterdata:
-            cached_query = cache_decorator(self.masterdata_cache)(
-                self._get_operational_entities
-            )
-            df = cached_query(table)
+    @property
+    def end_time(self) -> pd.Timestamp:
+        if self.as_of_time:
+            return self.as_of_time + pdz.get_epsilon(dtype=pd.DatetimeTZDtype(tz="UTC"))
         else:
-            df = self._get_operational_entities(table)
+            return self.time_interval.right
+
+    @staticmethod
+    def _intersect_time_and_comission_intervals(df: pd.DataFrame) -> pd.DataFrame:
+        df = df.copy()
+        # take max og start time and comission time (if latter is not null)
+        df["start_time_utc"] = df[["start_time_utc", "commission_time_utc"]].max(axis=1)
+        # take min of end time and decommission time (if latter is not null)
+        df["end_time_utc"] = df[["end_time_utc", "decommission_time_utc"]].min(axis=1)
+        # filter out rows where start time is after end time
+        df = df[df["start_time_utc"] <= df["end_time_utc"]]
 
+        df = df.drop(columns=["commission_time_utc", "decommission_time_utc"])
         return df
 
-    @typechecked
-    def get_data(
-        self,
-        table: str,
-        filters: dict = {},
-        columns: list = [],
-    ) -> pd.DataFrame:
-        """Gets the data for a given table.
-        Filters for rows valid at the end of time interval.
-        """
-        # Get operational entities
-        df = self.get_operational_entities(table)
-
-        # Apply the filters
-        for column, value in filters.items():
-            if isinstance(value, list):
-                df = df[df[column].isin(value)].reset_index()
-            else:
-                df = df[df[column] == value].reset_index()
-
-        for column in columns:
-            if column not in df.columns:
-                raise KeyError(f"Column {column} not found in {table}")
-
-        df = pdz.as_of_time(
-            df=df,
-            period=("valid_from_date_utc", "valid_to_date_utc"),
-            at=self.time_interval.right,
-        )
-        df = df.filter(columns)
+    def _filter_interval(self, df: pd.DataFrame) -> pd.DataFrame:
+        df["start_time_utc"] = df["start_time_utc"].clip(lower=self.start_time)
+        df["end_time_utc"] = df["end_time_utc"].clip(upper=self.end_time)
+        df = df[df["start_time_utc"] < df["end_time_utc"]]
+        return df
 
+    def _get_table(self, table_name: str) -> pd.DataFrame:
+        df = self.db.query(table_name)
+        df = df.drop(columns=["emda_version", "created_time_utc"], errors="ignore")
         return df
+
+    def get_plant_masterdata(self) -> pd.DataFrame:
+        df_plant = self._get_table("masterdata_emda_plants")
+        df_unit = self._get_table("masterdata_emda_units")
+        df_market_participant = self._get_table("masterdata_emda_market_participants")
+
+        df_plant = self._intersect_time_and_comission_intervals(df_plant)
+        df_unit = self._intersect_time_and_comission_intervals(df_unit)
+
+        df_plant = self._filter_interval(df_plant)
+        df_unit = self._filter_interval(df_unit)
+        df_market_participant = self._filter_interval(df_market_participant)
+
+        df_unit = df_unit.sort_values(by="capacity_max_MW", ascending=False)
+
+        df_unit_summary = pdz.group_interval(
+            df_unit,
+            by=["plant_id"],
+            interval=("start_time_utc", "end_time_utc"),
+        ).agg(
+            {
+                "power_system_resource_type": "first",  # From largest unit
+                "asset_type": "first",  # From largest unit
+                "power_system_resource": "first",  # From largest unit
+                "coordinate_x_utm": "mean",
+                "coordinate_y_utm": "mean",
+                "hub_height_m": "mean",
+                "c11": "first",  # Assuming unique per group
+            }
+        )
+
+        df_plant = pdz.merge_interval_interval(
+            df_plant,
+            df_unit_summary,
+            on=["plant_id"],
+            interval=("start_time_utc", "end_time_utc"),
+            how="left",
+        )
+
+        df_plant = pdz.merge_interval_interval(
+            df_plant,
+            df_market_participant,
+            on=["market_participant_id"],
+            interval=("start_time_utc", "end_time_utc"),
+            how="left",
+        )
+
+        df_address = self._get_table("masterdata_address")
+        df_address = df_address.rename(
+            columns={
+                "street_name": "address_street_name",
+                "house_number": "address_house_number",
+                "postal_code": "address_postal_code",
+                "latitude": "address_latitude",
+                "longitude": "address_longitude",
+            }
+        )
+
+        df_plant = pdz.merge(
+            df_plant,
+            df_address,
+            on=["address_street_name", "address_house_number", "address_postal_code"],
+            how="left",
+        )
+
+        df_plant["is_household"] = is_household(df_plant)
+
+        df_plant["latitude"], df_plant["longitude"] = coordinates(df_plant)
+        df_plant = df_plant.drop(columns=["address_latitude", "address_longitude"])
+
+        return df_plant
+
+
+def is_household(df: pd.DataFrame) -> pd.Series:
+    """
+    Determine if a plant type corresponds to a household.
+    """
+    # due to inadequate quality of master data
+    # we apply several filters to estimate
+    # if an installation is a household or not,
+    # based on the description found in
+    # https://ens.dk/sites/ens.dk/files/Stoette_vedvarende_energi/energistyrelsens_vejledning_om_beregning_af_nettoafregning_og_opgoerelse_.pdf
+
+    # household installations will be behind-the-meter
+    # ("installationstilsluttet")
+    is_behind_the_meter = df["connection_point"] == CONNECTION_POINT_BEHIND_THE_METER
+
+    # household installations will be in the yearly
+    # settlement group 6 ("årsbaseret nettoafregning")
+    is_settlement_group_6 = df["settlement_group"] == 6
+
+    # also remove installations with installed power
+    # below 200 KW, as these will also most likely
+    # be behind-the-meter installations (this should be
+    # captured already in the above filters, but
+    # masterdata is not fully reliable)
+    # an example of a 200 KW plant can be found at coordinates 55.691, 9.397
+    is_small = df["capacity_max_MW"] <= 0.2
+
+    # if the installation is connected to the TSO,
+    # it is definitely not a household installation
+    is_tso_connected = df["operation_type"] == "Tso"
+
+    return (is_behind_the_meter | is_settlement_group_6 | is_small) & ~is_tso_connected
+
+
+def coordinates(
+    df: pd.DataFrame,
+) -> pd.DataFrame:
+    # Translate UTM to lat/lon if they exist
+    latitude, longitude = utm_to_latlon(
+        df["coordinate_x_utm"],
+        df["coordinate_y_utm"],
+        df["price_area"],
+    )
+
+    # Prefer translated utm coordinates, fall back to address coordinates
+    latitude = latitude.combine_first(df["address_latitude"])
+    longitude = longitude.combine_first(df["address_longitude"])
+
+    return latitude, longitude
+
+
+def utm_to_latlon(
+    x: pd.Series, y: pd.Series, price_area: pd.Series
+) -> tuple[pd.Series, pd.Series]:
+    """
+    Convert UTM coordinates to latitude and longitude, given the price area.
+    Ideally, we would use the zone number, but this is not available in masterdata
+    currently. Instead, we use a workaround based on the price area, which should
+    be sufficient for now. It follows the following logic:
+    - If the price area is DK1, use UTM zone 32
+    - If the price area is DK2, use UTM zone 33, unless the resulting longitude
+      is above 16.0 (about 50 km to the right of Bornholm): In this case, we assume
+      the real zone is actually 32, since no entities should be placed there.
+    - If the price area is undefined, use UTM zone 32, since this is the majority of
+      the area in Denmark.
+    """
+    if x.empty or y.empty or x.isnull().all() or y.isnull().all():
+        return pd.Series(dtype=float), pd.Series(dtype=float)
+
+    lat_32, lon_32 = utm.to_latlon(x, y, zone_number=32, northern=True, strict=False)
+
+    lat_33, lon_33 = utm.to_latlon(x, y, zone_number=33, northern=True, strict=False)
+
+    in_DK1 = price_area == "DK1"
+    in_DK2 = price_area == "DK2"
+    undefined = price_area.isnull()
+
+    is_zone_32 = in_DK1 | undefined | (in_DK2 & (lon_33 >= 16.0))
+
+    lat = lat_32.where(is_zone_32, lat_33)
+    lon = lon_32.where(is_zone_32, lon_33)
+
+    return lat, lon

domainiac/managers/schemas/masterdata_manager.json

@@ -0,0 +1,40 @@
+{
+  "address_city": "object",
+  "address_country": "object",
+  "address_house_number": "object",
+  "address_municipality": "object",
+  "address_postal_code": "object",
+  "address_street_name": "object",
+  "asset_type": "object",
+  "c11": "object",
+  "capacity_max_MW": "float64",
+  "capacity_min_MW": "float64",
+  "connection_point": "object",
+  "coordinate_x_utm": "float64",
+  "coordinate_y_utm": "float64",
+  "datahub_gsrn_e17": "object",
+  "datahub_gsrn_e18": "object",
+  "end_time_utc": "datetime64[ns, UTC]",
+  "hub_height_m": "float64",
+  "market_participant_ecp_eic": "object",
+  "market_participant_eic": "object",
+  "market_participant_gln": "object",
+  "market_participant_id": "object",
+  "market_participant_name": "object",
+  "market_participant_short_name": "object",
+  "operation_type": "object",
+  "plant_gsrn": "object",
+  "plant_id": "object",
+  "plant_name": "object",
+  "plant_short_name": "object",
+  "plant_type": "object",
+  "power_system_resource": "object",
+  "power_system_resource_type": "object",
+  "price_area": "object",
+  "settlement_group": "object",
+  "start_time_utc": "datetime64[ns, UTC]",
+  "substation_id": "object",
+  "is_household": "bool",
+  "latitude": "float64",
+  "longitude": "float64"
+}

domainiac/modeling/plant.py

@@ -12,7 +12,7 @@ class Plant:
     datahub_gsrn_e18: str
     price_area: str
     coordinate: Coordinate
-    installed_power_MW: float
+    capacity_max_MW: float
     power_system_resource_type: str
 
     @classmethod
@@ -40,7 +40,7 @@ class Plant:
                     longitude=row["longitude"],
                     altitude=altitude,
                 ),
-                installed_power_MW=row["installed_power_MW"],
+                capacity_max_MW=row["capacity_max_MW"],
                 power_system_resource_type=psrt,
             )
             plants.append(plant)
@@ -50,7 +50,7 @@ class Plant:
 @dataclass(frozen=True)
 class Group:
     coordinate: Coordinate
-    installed_power_MW: float
+    capacity_max_MW: float
     identifiers: dict[str, str]
 
     @classmethod
@@ -65,7 +65,7 @@ class Group:
             identifiers = {identifier: row[identifier] for identifier in identifiers}
             group = cls(
                 identifiers=identifiers,
-                installed_power_MW=row["installed_power_MW"],
+                capacity_max_MW=row["capacity_max_MW"],
                 coordinate=Coordinate(
                     latitude=row["latitude"],
                     longitude=row["longitude"],

{domainiac-9.3.1.dist-info → domainiac-10.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: domainiac
-Version: 9.3.1
+Version: 10.0.0
 Summary: Package for working with Energinet data, but with specialized functions used for Enigma.
 Author: Team Enigma
 Author-email: enigma@energinet.dk
@@ -17,3 +17,4 @@ Requires-Dist: pvlib (>=0.13.1)
 Requires-Dist: scikit-learn (>=1.3.0)
 Requires-Dist: scipy (>=1.15.3)
 Requires-Dist: typeguard (>=4.2.1)
+Requires-Dist: utm (>=0.8.1)

{domainiac-9.3.1.dist-info → domainiac-10.0.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-domainiac/__init__.py,sha256=Xaeq4K8dcStlSilXiL6Sr0fCNgtPgWCKi93n8Cu8Cjw,232
+domainiac/__init__.py,sha256=LF5tqnpDkIGcy5fsqLGJ1Yg1GeWdpI0BC0QYjz2xpgc,220
 domainiac/functions/__init__.py,sha256=ZHNbj1aWRnUgM6TAlQqGLv2r-GdAvoaTrPIin1w5Dk0,137
 domainiac/functions/conversions.py,sha256=RZU5cpHD4uBHeqFfiFplJCv5gN-xzEHvQCKrizi1hMA,490
 domainiac/functions/interpolation.py,sha256=qj-sTEbez_IcXLY7eOS_KqAAd8GbhP5JBkdWrUPjWIQ,3720
@@ -6,20 +6,19 @@ domainiac/functions/solar.py,sha256=nU9ivDLA4gE0CTMttCb_7ztoQUQicWNe_11ZqVC7d-Y,
 domainiac/functions/temperature.py,sha256=-t_zWYIhYwJpyZEQgwxgR9SkeYAAjvb6qeV8LtDtfmk,940
 domainiac/functions/typing.py,sha256=gOvHUYIWSEv_aMttY1nMEyh-pdQKMJ8O31x9y47-U5U,229
 domainiac/functions/wind.py,sha256=ZOxSMxjWV7LZsAQVW5Y-GPMhZzfvIsh3NOtSXB5lUAY,993
-domainiac/managers/__init__.py,sha256=A4KdK_LiukXKlJLIF1hf8HF7xh3QeFdBqAWstXc8CgI,352
+domainiac/managers/__init__.py,sha256=rWg7CX8Y7iFHZqt5zVgUccNeBmdSSmamOXfcv4pbxak,274
 domainiac/managers/availability_manager.py,sha256=zKRjZlG5wp3MlviIbNg3s5da9nUKjP7_r8R5g8hHRg4,1895
-domainiac/managers/masterdata_manager.py,sha256=_xIHEWVYssdvC2VMujvR0ANXCI_Er3qOlfAX427yimk,2298
+domainiac/managers/masterdata_manager.py,sha256=0D71Tol5ij-4kljlRyz8xR0xDVNyTBmzQQmrLaSoiGE,8524
 domainiac/managers/metering_manager.py,sha256=MauMncY_yaVPx8Y9CCD8nJiE1BGHFKTVC1bOtiyl2pI,1701
 domainiac/managers/nwp_manager.py,sha256=h2J6ryXqplB_W6jC7aSenOa9MOxsYrd2HenU3WZnyu0,10040
 domainiac/managers/outage_manager.py,sha256=KnBjsDFX8xQvA2UGh7gX2fTVwoQlF6PFoy5Uk_s3kfs,1436
-domainiac/managers/plant_manager.py,sha256=Bh1jPwepDq_yO29QVZndjCXe3ScQndIsKwFZnjr9a8w,5101
 domainiac/managers/resource_manager.py,sha256=QMkNcrgyXLLlkCR9kg4-nFjTaYHtlsldex-b_8lAmO0,3730
-domainiac/managers/unit_manager.py,sha256=DYF-AiBHzUGZJE5XlTLcszvgIQkSzZs_s9o35l4bkVk,1219
+domainiac/managers/schemas/masterdata_manager.json,sha256=GjLvlcO_TK5CC0BQ64uXLmBMgSPdF0f___ekjmeZ3BE,1234
 domainiac/modeling/__init__.py,sha256=Ac5feYZ9gVLayU1Cb1ACqOyDrmvcxoOvZuUkOeXVKf8,101
 domainiac/modeling/nwp.py,sha256=PolrBdQn8W-e1M0_pefvmLn2mr4HT0NqlYlkyCV0dds,438
-domainiac/modeling/plant.py,sha256=Y0_Q6V6Lj3irJh5z-49r5gFJdD4Xl_pfHIMUhdaJVUI,2226
+domainiac/modeling/plant.py,sha256=ay1Tlyjn4Flto8n2Wv8DbpLxCBMj7IpfR4uUcvvaea0,2208
 domainiac/wrappers/__init__.py,sha256=vZOw9maXgVoAvudZqioD-GTPkgPI6fm7_CUELQcR_-g,43
 domainiac/wrappers/cache_wrapper.py,sha256=jDg-Lt_y-YzItyP-74tGULOxb07s_CcutmOHP1Ie00Q,355
-domainiac-9.3.1.dist-info/METADATA,sha256=MonbkoqmH-AXby94IcR3J04hcDP9wiqeD3EUEM3hWlA,737
-domainiac-9.3.1.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-domainiac-9.3.1.dist-info/RECORD,,
+domainiac-10.0.0.dist-info/METADATA,sha256=UqOyPiUJ3Fbb_2mFIc61VtIzUFOEMj1KtuZXGsWlKIk,767
+domainiac-10.0.0.dist-info/WHEEL,sha256=kJCRJT_g0adfAJzTx2GUMmS80rTJIVHRCfG0DQgLq3o,88
+domainiac-10.0.0.dist-info/RECORD,,

{domainiac-9.3.1.dist-info → domainiac-10.0.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.3.0
+Generator: poetry-core 2.3.1
 Root-Is-Purelib: true
 Tag: py3-none-any

domainiac/managers/plant_manager.py

@@ -1,160 +0,0 @@
-import datamazing.pandas as pdz
-import pandas as pd
-
-from .masterdata_manager import MasterdataManager
-from .unit_manager import UnitManager
-
-
-class PlantManager(MasterdataManager):
-    """
-    Manager which simplifies the process of getting plants from masterdata.
-    """
-
-    def __init__(
-        self,
-        db: pdz.Database,
-        time_interval: pdz.TimeInterval,
-        resolution: pd.Timedelta,
-        cache_masterdata: bool = False,
-    ) -> None:
-        self.db = db
-        self.time_interval = time_interval
-        self.resolution = resolution
-        self.cache_masterdata = cache_masterdata
-        self.unit_manager = UnitManager(db, time_interval, resolution, cache_masterdata)
-
-    def get_plants(
-        self,
-        filters: dict = {},
-        columns: list | None = None,
-    ) -> pd.DataFrame:
-        """Gets the plants for a given plant type.
-        Filters for plants valid at the end of time interval.
-        Filters by default for plants in operation.
-        """
-        default_columns = [
-            "plant_id",
-            "masterdata_gsrn",
-            "datahub_gsrn_e18",
-            "installed_power_MW",
-            "price_area",
-            "is_tso_connected",
-            "valid_from_date_utc",
-            "valid_to_date_utc",
-            "primary_net_component_id",
-        ]
-        if not columns:
-            columns = default_columns
-
-        # TODO: masterdata_plant table doesn't have net_component_id column
-        # Find a better way to do this in future.
-        plant_columns = [col for col in columns if col != "primary_net_component_id"]
-        df_plant = self.get_data(
-            "masterdataPlant", filters=filters, columns=plant_columns
-        )
-        df_psr = self._get_power_system_resource()
-        df = df_plant.merge(
-            df_psr, on=["plant_id"], how="left", validate="m:1"
-        ).drop_duplicates()
-
-        df = df[columns]
-
-        return df
-
-    def get_installed_power_timeseries(self, gsrn: str) -> pd.DataFrame:
-        """Gets the installed power timeseries for a plant."""
-
-        df_times = self.time_interval.to_range(self.resolution).to_frame(
-            index=False, name="time_utc"
-        )
-
-        # explode plant to time series
-        df_plant = self.get_operational_entities("masterdataPlant")
-        df_plant = df_plant.query(f"masterdata_gsrn == '{gsrn}'")
-
-        df_plant = pdz.merge_point_interval(
-            df_times,
-            df_plant,
-            left_point="time_utc",
-            right_interval=("valid_from_date_utc", "valid_to_date_utc"),
-        )
-
-        return df_plant.filter(["time_utc", "installed_power_MW"]).reset_index(
-            drop=True
-        )
-
-    def _get_corrected_installed_power(
-        self, gsrn: str, df_invalid_periods: pd.DataFrame
-    ):
-        df_times = self.time_interval.to_range(self.resolution).to_frame(
-            index=False, name="time_utc"
-        )
-        df = self.get_installed_power_timeseries(gsrn=gsrn)
-
-        # explode invalid periods to time series
-        df_invalid_periods = df_invalid_periods.query(f"masterdata_gsrn == '{gsrn}'")
-        df_invalid_periods = pdz.merge(
-            df_times,
-            df_invalid_periods,
-            left_time="time_utc",
-            right_period=("start_date_utc", "end_date_utc"),
-        )
-
-        df = pdz.merge(
-            df,
-            df_invalid_periods,
-            on="time_utc",
-            how="left",
-        )
-
-        # correct installed power for invalid periods
-        df["installed_power_MW"] = df["installed_power_MW"].where(
-            df["corrected_installed_power_MW"].isnull(),
-            df["corrected_installed_power_MW"],
-        )
-
-        df = df[["time_utc", "installed_power_MW"]]
-
-        return df
-
-    def _get_power_system_resource(self) -> pd.DataFrame:
-
-        df_unit = self.unit_manager.get_units(
-            columns=["masterdata_gsrn", "capacity_min_MW", "capacity_max_MW"]
-        )
-
-        df_psr_mapping = self.db.query("masterdataAggregatedUnit")[
-            ["unit_gsrn", "net_component_id"]
-        ]
-
-        df = pd.merge(
-            df_psr_mapping,
-            df_unit,
-            left_on="unit_gsrn",
-            right_on="masterdata_gsrn",
-            how="left",
-            validate="1:m",
-        )
-
-        # for a small number of plants, the underlying unit can
-        # be associated with different net components. Too avoid
-        # this issue, we choose for each plant, the net component,
-        # for which the underlying units amounts to the largest
-        # capacity
-        df = pdz.group(df, by=["net_component_id", "plant_id"]).agg(
-            {"capacity_min_MW": "sum", "capacity_max_MW": "sum"}
-        )
-
-        df["capacity_range_MW"] = df["capacity_max_MW"] - df["capacity_min_MW"]
-
-        df = df.sort_values(
-            ["plant_id", "capacity_range_MW"], ascending=False
-        ).drop_duplicates(subset=["plant_id"], keep="first")
-
-        df = df.rename(
-            columns={
-                "net_component_id": "primary_net_component_id",
-            }
-        )
-
-        return df

domainiac/managers/unit_manager.py

@@ -1,42 +0,0 @@
-import datamazing.pandas as pdz
-import pandas as pd
-
-from .masterdata_manager import MasterdataManager
-
-
-class UnitManager(MasterdataManager):
-    """
-    Manager which simplifies the process of getting units from masterdata.
-    """
-
-    def __init__(
-        self,
-        db: pdz.Database,
-        time_interval: pdz.TimeInterval,
-        resolution: pd.Timedelta,
-        cache_masterdata: bool = False,
-    ) -> None:
-        self.db = db
-        self.time_interval = time_interval
-        self.resolution = resolution
-        self.cache_masterdata = cache_masterdata
-
-    def get_units(
-        self,
-        filters: dict = {},
-        columns: list | None = None,
-    ) -> pd.DataFrame:
-        """Gets the units for a given unit type.
-        Filters for units valid at the end of time interval.
-        Filters by default for units in operation.
-        """
-        default_columns = [
-            "masterdata_gsrn",
-            "plant_id",
-            "power_system_resource_type",
-        ]
-        if not columns:
-            columns = default_columns
-        else:
-            columns = list(set(default_columns + columns))
-        return self.get_data("masterdataUnit", filters=filters, columns=columns)