Skip to content

Commit a99b46d

Browse files
committed
Cleanup naming and remove tf dependency
1 parent 357a513 commit a99b46d

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

pyoneer/rl/strategies/mode_strategy_impl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,5 @@ def __init__(self, policy):
1515
self.policy = policy
1616

1717
def __call__(self, *args, **kwargs):
18-
policy = self.policy(*args, **kwargs)
19-
return policy.mode()
18+
dist = self.policy(*args, **kwargs)
19+
return dist.mode()

pyoneer/rl/strategies/sample_strategy_impl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,5 @@ def __init__(self, policy):
1515
self.policy = policy
1616

1717
def __call__(self, *args, **kwargs):
18-
policy = self.policy(*args, **kwargs)
19-
return policy.sample()
18+
dist = self.policy(*args, **kwargs)
19+
return dist.sample()

setup.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66

77
REQUIRED_PACKAGES = [
88
'gym',
9-
'tensorflow',
109
'tensorflow-probability',
1110
]
1211

0 commit comments

Comments
 (0)