Added PolicyEngine Core installation and testing

PolicyEngine · Jun 28, 2024 · e2f127c · e2f127c
1 parent 548625b
commit e2f127c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/reweight/tests/test_installation.py b/reweight/tests/test_installation.py
@@ -2,6 +2,7 @@ def test_imports():
     try:
         import torch
         import numpy as np
+        import policyengine_core
         from torch.utils.tensorboard import SummaryWriter
     except:
         raise AssertionError("Failed to import necessary libraries")

diff --git a/setup.py b/setup.py
@@ -32,6 +32,7 @@
         "tensorboard",
         "jupyter-book",
         "pytest",
+        "policyengine-core",
     ],
     extras_require={
         "dev": [