Change references from distribution_strategy_context.py to `distrib…

…ute_lib.py` in preparation for combining them into `distribute_lib.py`.

PiperOrigin-RevId: 527412843

keras/distribute/distributed_file_utils.py

@@ -84,7 +84,7 @@ def write_dirpath(dirpath, strategy):
       The writing dir path that should be used to save with distribution.
     """
     if strategy is None:
-        # Infer strategy from `distribution_strategy_context` if not given.
+        # Infer strategy from `tf.distribute` if not given.
         strategy = tf.distribute.get_strategy()
     if strategy is None:
         # If strategy is still not available, this is not in distributed
@@ -107,7 +107,7 @@ def remove_temp_dirpath(dirpath, strategy):
       strategy: The tf.distribute strategy object currently used.
     """
     if strategy is None:
-        # Infer strategy from `distribution_strategy_context` if not given.
+        # Infer strategy from `tf.distribute` if not given.
         strategy = tf.distribute.get_strategy()
     if strategy is None:
         # If strategy is still not available, this is not in distributed