From fe7c7b3db0ec639fc761f7c43f3603d2fe6bddff Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Wed, 22 Jan 2025 13:45:34 +0100
Subject: [PATCH] better handling of missing cpp_projection

---
 fancy_rl/projections/kl_projection.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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