Merge branch 'main' of github.com:llnl/abmarl into abmarl-456-sim-unw…

…rapped-in-stage
LLNL · Feb 7, 2024 · 01ce046 · 01ce046
2 parents 695e18e + f3f4915
commit 01ce046
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml
@@ -26,5 +26,7 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install -r requirements/requirements_all.txt
+    - name: Install Abmarl
+      run: pip install . --no-deps
     - name: Test with pytest
       run: pytest
diff --git a/abmarl/stage.py b/abmarl/stage.py
@@ -158,12 +158,12 @@ def animate(i):
         )
         if record:
             anim.save(
-                os.path.join(params['ray_tune']['local_dir'], 'Episode_{}.gif'.format(episode))
+                os.path.join(full_trained_directory, 'Episode_{}.gif'.format(episode))
             )
             plt.show(block=False)
         elif record_only:
             anim.save(
-                os.path.join(params['ray_tune']['local_dir'], 'Episode_{}.gif'.format(episode))
+                os.path.join(full_trained_directory, 'Episode_{}.gif'.format(episode))
             )
 
         while not all_done: