Internal change.

PiperOrigin-RevId: 553184312 Change-Id: I3de6dcbd8289d0726f8714b84b166645a0e297a1
tensorflow · Aug 2, 2023 · 6c32f58 · 6c32f58
1 parent da4be6a
commit 6c32f58
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/tf_agents/bandits/environments/bandit_py_environment.py b/tf_agents/bandits/environments/bandit_py_environment.py
@@ -116,5 +116,5 @@ def _observe(self) -> types.NestedArray:
     """Returns an observation."""
 
   @property
-  def name(self) -> Text:
+  def name(self) -> Optional[Text]:
     return self._name
diff --git a/tf_agents/bandits/environments/bandit_tf_environment.py b/tf_agents/bandits/environments/bandit_tf_environment.py
@@ -146,5 +146,5 @@ def _observe(self) -> types.NestedTensor:
     """Returns an observation."""
 
   @property
-  def name(self) -> Text:
+  def name(self) -> Optional[Text]:
     return self._name
diff --git a/tf_agents/bandits/environments/piecewise_bernoulli_py_environment.py b/tf_agents/bandits/environments/piecewise_bernoulli_py_environment.py
@@ -116,7 +116,7 @@ def __init__(
     )
 
   @property
-  def batch_size(self) -> int:
+  def batch_size(self) -> Optional[int]:
     return self._batch_size
 
   @property

diff --git a/tf_agents/bandits/environments/ranking_environment.py b/tf_agents/bandits/environments/ranking_environment.py
@@ -196,7 +196,7 @@ def batched(self) -> bool:
     return True
 
   @property
-  def batch_size(self) -> int:
+  def batch_size(self) -> Optional[int]:
     return self._batch_size
 
   def _observe(self) -> types.NestedArray:

diff --git a/tf_agents/bandits/environments/stationary_stochastic_per_arm_py_environment.py b/tf_agents/bandits/environments/stationary_stochastic_per_arm_py_environment.py
@@ -136,7 +136,7 @@ def batched(self) -> bool:
     return True
 
   @property
-  def batch_size(self) -> int:
+  def batch_size(self) -> Optional[int]:
     return self._batch_size
 
   def _observe(self) -> types.NestedArray:

diff --git a/tf_agents/bandits/environments/stationary_stochastic_py_environment.py b/tf_agents/bandits/environments/stationary_stochastic_py_environment.py
@@ -116,7 +116,7 @@ def batched(self) -> bool:
     return True
 
   @property
-  def batch_size(self) -> int:
+  def batch_size(self) -> Optional[int]:
     return self._batch_size
 
   def _observe(self) -> types.NestedArray:

diff --git a/tf_agents/bandits/environments/stationary_stochastic_structured_py_environment.py b/tf_agents/bandits/environments/stationary_stochastic_structured_py_environment.py
@@ -128,7 +128,7 @@ def batched(self) -> bool:
     return True
 
   @property
-  def batch_size(self) -> int:
+  def batch_size(self) -> Optional[int]:
     return self._batch_size
 
   def _generate_batch_of_observations(self, generator_fn, num_samples):