Cleanup naming and remove tf dependency

fomorians-oss · Feb 12, 2019 · a99b46d · a99b46d
1 parent 357a513
commit a99b46d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/pyoneer/rl/strategies/mode_strategy_impl.py b/pyoneer/rl/strategies/mode_strategy_impl.py
@@ -15,5 +15,5 @@ def __init__(self, policy):
         self.policy = policy
 
     def __call__(self, *args, **kwargs):
-        policy = self.policy(*args, **kwargs)
-        return policy.mode()
+        dist = self.policy(*args, **kwargs)
+        return dist.mode()
diff --git a/pyoneer/rl/strategies/sample_strategy_impl.py b/pyoneer/rl/strategies/sample_strategy_impl.py
@@ -15,5 +15,5 @@ def __init__(self, policy):
         self.policy = policy
 
     def __call__(self, *args, **kwargs):
-        policy = self.policy(*args, **kwargs)
-        return policy.sample()
+        dist = self.policy(*args, **kwargs)
+        return dist.sample()
diff --git a/setup.py b/setup.py
@@ -6,7 +6,6 @@
 
 REQUIRED_PACKAGES = [
     'gym',
-    'tensorflow',
     'tensorflow-probability',
 ]