[RLlib] Minor cleanup in preparation to tf2.x support. (#9130)

* WIP. * Fixes. * LINT. * Fixes. * Fixes and LINT. * WIP.
2026-06-28 04:23:03 +08:00 · 2020-06-25 19:01:32 +02:00
parent aa3fd62cac
commit 4fd8977eaf
37 changed files with 347 additions and 176 deletions
@@ -42,8 +42,8 @@ class TensorFlowVariables:
        Args:
            output (tf.Operation, List[tf.Operation]): The tensorflow
                operation to extract all variables from.
-            sess (tf.Session): Session used for running the get and set
-                methods.
+            sess (Optional[tf.Session]): Optional tf.Session used for running
+                the get and set methods in tf graph mode.
            input_variables (List[tf.Variables]): Variables to include in the
                list.
        """