fix relative imports

nens · Nov 29, 2024 · 526ba27 · 526ba27
1 parent 614b19d
commit 526ba27
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/utils/threedi_result_aggregation/base.py b/utils/threedi_result_aggregation/base.py
@@ -19,14 +19,14 @@
 from osgeo import osr
 
 
-from constants import (
+from .constants import (
     AGGREGATION_VARIABLES,
     NON_TS_REDUCING_KCU,
     NP_OGR_DTYPES,
     EXCHANGE_LEVEL_1D2D,
     THRESHOLD_PRECISION,
 )
-from aggregation_classes import (
+from .aggregation_classes import (
     Aggregation,
     AggregationSign,
     AggregationMethod,
@@ -39,7 +39,7 @@
     VT_NODE_HYBRID,
     VT_PUMP
 )
-from threedigrid_ogr import threedigrid_to_ogr
+from .threedigrid_ogr import threedigrid_to_ogr
 
 warnings.filterwarnings("ignore")
 ogr.UseExceptions()
@@ -353,8 +353,7 @@ def equal(values: np.array, threshold: Union[float, np.array], precision=THRESHO
         "time_above_threshold": greater,
     }
 
-    if start_time is None:
-        start_time = 0
+    start_time = start_time or 0
 
     if aggregation.method.short_name == "sum":
         raw_values_per_time_interval = np.multiply(timeseries.T, tintervals).T

diff --git a/utils/threedi_result_aggregation/constants.py b/utils/threedi_result_aggregation/constants.py
@@ -6,7 +6,7 @@
 from threedigrid.admin.utils import KCUDescriptor
 
 # local
-from aggregation_classes import (
+from .aggregation_classes import (
     AggregationVariableList,
     AggregationSign,
     AggregationMethod,