input_options should be popped instead of using `get`.

PiperOrigin-RevId: 405317879

orbit/utils/common.py

@@ -64,7 +64,7 @@ def make_distributed_dataset(strategy, dataset_or_fn, *args, **kwargs):
   if strategy is None:
     strategy = tf.distribute.get_strategy()
 
-  input_options = kwargs.get("input_options", None)
+  input_options = kwargs.pop("input_options", None)
 
   if isinstance(dataset_or_fn, tf.data.Dataset):
     return strategy.experimental_distribute_dataset(dataset_or_fn,