deploy AT

.github/workflows/basic_AT.yml

@@ -28,6 +28,11 @@ jobs:
 
     - name: Install temps
       run: pip install -e .[explicit]
+    # Add this new step to download model files
+    - name: Download model files
+      run: |
+        mkdir -p data/models
+        wget -O data/models/modelF_DA.pt https://raw.githubusercontent.com/TEMPS/master/data/models/modelF_DA.pt
 
     # Run the tests and specify the temporary directory as output_dir
     - name: Run tests

tests/test_temps.py

@@ -17,7 +17,7 @@ def test():
 
     col = np.array([0.54804805, 1.81142339, 0.63354394, 0.7356338 , 1.3578122 ,
        0.90108565])
-    ztrue = 0.444
+    ztrue = 0.4446
 
     z, pz, odds = temps_module.get_pz(input_data=torch.Tensor(col), 
                                             return_pz=True,