fix: properties

21cmfast · Jul 16, 2024 · d3efecd · d3efecd
1 parent 1a13703
commit d3efecd
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 2 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -10,3 +10,4 @@ sphinx>=5.2.0
 sphinx-click>=4.4.0
 tensorflow>=2.6.0
 toml==0.10.2
+torch>=1.9.0
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,6 +35,7 @@ click = ">=8.0.1"
 numpy = "^1.22.0"
 scipy = "^1.10.1"
 tensorflow = ">=2.4.0, <= 2.14.0"
+torch = ">=1.9.0"
 appdirs = "^1.4.4"
 toml = "^0.10.2"
 GitPython = "^3.1.31"

diff --git a/...mu/models/radio_background/FINAL_EMULATOR → ...o_background/Radio_Background_Emu_Weights b/...mu/models/radio_background/FINAL_EMULATOR → ...o_background/Radio_Background_Emu_Weights
diff --git a/src/py21cmemu/models/radio_background/median_test_errors.npz b/src/py21cmemu/models/radio_background/median_test_errors.npz
diff --git a/src/py21cmemu/models/radio_background/radio_background_emu_csts.npz b/src/py21cmemu/models/radio_background/radio_background_emu_csts.npz
diff --git a/src/py21cmemu/properties.py b/src/py21cmemu/properties.py
@@ -99,7 +99,9 @@ class RadioBackgroundEmulatorProperties(EmulatorProperties):
 
     def __init__(self):
         here = Path(__file__).parent
-        all_emulator_numbers = np.load(here / "models/radio_background/Feb_wPScsts.npz")
+        all_emulator_numbers = np.load(
+            here / "models/radio_background/radio_background_emu_csts.npz"
+        )
         self._data = all_emulator_numbers
 
         self.logPS_mean = all_emulator_numbers["logPS_mean"]
@@ -115,7 +117,7 @@ def __init__(self):
         self.logTr_std = all_emulator_numbers["logTr_std"]
 
         self.mean_errors = np.load(
-            "/home/dbreitman/Radio_Background/Models/Final_model/median_test_errors.npz"
+            here / "models/radio_background/median_test_errors.npz"
         )