Add Identity variation class that simply returns current value.

sbohez · copybara-github · commit 1f8629dd0a68 · 2022-02-07T05:11:40.000-08:00
PiperOrigin-RevId: 426888877
Change-Id: I25865a8629c4d750b1ba52331b87a92f4cbb6b9f
diff --git a/dm_control/composer/variation/deterministic.py b/dm_control/composer/variation/deterministic.py
@@ -46,3 +46,9 @@ def __call__(self, initial_value=None, current_value=None, random_state=None):
     except StopIteration:
       self._iterator = iter(self._values)
       return next(self._iterator)
+
+
+class Identity(base.Variation):
+
+  def __call__(self, initial_value=None, current_value=None, random_state=None):
+    return current_value
diff --git a/setup.py b/setup.py
@@ -192,7 +192,7 @@ def is_excluded(s):
 
 setup(
     name='dm_control',
-    version='0.0.425478996',
+    version='0.0.426888877',
     description='Continuous control environments and MuJoCo Python bindings.',
     author='DeepMind',
     license='Apache License, Version 2.0',
