better handling of missing cpp_projection

2025-01-22 13:45:34 +01:00 · 2025-01-22 13:45:34 +01:00 · fe7c7b3db0
commit fe7c7b3db0
parent 90666a695c
1 changed files with 5 additions and 1 deletions
--- a/fancy_rl/projections/kl_projection.py
+++ b/fancy_rl/projections/kl_projection.py
@ -1,5 +1,9 @@
 import torch
-import cpp_projection
+try:
+    import cpp_projection
+    cpp_projection_available = True
+except ImportError:
+    cpp_projection_available = False
 import numpy as np
 from .base_projection import BaseProjection
 from tensordict.nn import TensorDictModule