BuildingEnv: fix imports for stochastic generation

chrisyeh96 · May 29, 2024 · 1856aca · 1856aca
1 parent 820a5b9
commit 1856aca
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 2 deletions.
diff --git a/...nvs/building/test_stochastic_generator.py → ...les/building/test_stochastic_generator.py b/...nvs/building/test_stochastic_generator.py → ...les/building/test_stochastic_generator.py
diff --git a/...ng/stochastic_uncontrollable_generator.py → ...gym/envs/building/stochastic_generator.py b/...ng/stochastic_uncontrollable_generator.py → ...gym/envs/building/stochastic_generator.py
diff --git a/sustaingym/envs/building/utils.py b/sustaingym/envs/building/utils.py
@@ -37,8 +37,7 @@
 from scipy import interpolate
 
 from sustaingym.data.utils import read_to_stringio
-
-from stochastic_uncontrollable_generator import StochasticUncontrollableGenerator
+from sustaingym.envs.building.stochastic_generator import StochasticUncontrollableGenerator
 
 
 class Ufactor(NamedTuple):