fix: loading vessel_subset with id

dataforgoodfr · Mar 29, 2024 · ed1294d · ed1294d
1 parent fd1f1ff
commit ed1294d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/tasks/dimensions/load_dim_vessel_from_csv.py b/src/tasks/dimensions/load_dim_vessel_from_csv.py
@@ -14,6 +14,7 @@ class LoadDimVesselFromCsv(BaseTask):
     def map_to_domain(self, row: pd.Series) -> Vessel:
         isna = row.isna()
         return Vessel(
+            id=int(row['id']),
             mmsi=int(row["mmsi"]) if not isna["mmsi"] else None,
             ship_name=row["ship_name"],
             width=None,