diff --git a/fancy_rl/projections/kl_projection.py b/fancy_rl/projections/kl_projection.py index eb65fd8..cd1a5fb 100644 --- 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