Fix module import in tests

iiasa · Nov 16, 2023 · 440e4ae · 440e4ae
1 parent 04b2254
commit 440e4ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/message_ix_models/tests/model/water/test_cooling.py b/message_ix_models/tests/model/water/test_cooling.py
@@ -2,7 +2,7 @@
 
 import pandas as pd
 
-from message_ix_models.model.water.data.irrigation import non_cooling_tec
+from message_ix_models.model.water.data.water_supply import non_cooling_tec
 
 
 def test_non_cooling_tec():

diff --git a/message_ix_models/tests/model/water/test_utils.py b/message_ix_models/tests/model/water/test_utils.py
@@ -3,7 +3,7 @@
 import pandas as pd
 import xarray as xr
 
-from message_ix_models.model.water import (
+from message_ix_models.model.water.build import (
     Context,
     add_commodity_and_level,
     map_add_on,