dodox/fancy_rl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

better handling of missing cpp_projection

Browse Source
This commit is contained in:
Dominik Moritz Roth 2025-01-22 13:45:34 +01:00
parent 90666a695c
commit fe7c7b3db0
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fancy_rl/projections/kl_projection.py
View File

@ -1,5 +1,9 @@
import torch import torch
import cpp_projection try:
import cpp_projection
cpp_projection_available = True
except ImportError:
cpp_projection_available = False
import numpy as np import numpy as np
from .base_projection import BaseProjection from .base_projection import BaseProjection
from tensordict.nn import TensorDictModule from tensordict.nn import TensorDictModule