better handling of missing cpp_projection

2025-01-22 13:45:34 +01:00
parent 90666a695c
commit fe7c7b3db0
1 changed files with 5 additions and 1 deletions
@@ -1,5 +1,9 @@
 import torch
+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