fix: post ray 2.52.1 fix for mnist script

Author: kryanbeane

demo-notebooks/guided-demos/mnist_fashion.py

@@ -8,15 +8,6 @@
 from ray.train import ScalingConfig
 
 
-def get_dataset():
-    return datasets.FashionMNIST(
-        root="/tmp/data",
-        train=True,
-        download=True,
-        transform=ToTensor(),
-    )
-
-
 class NeuralNetwork(nn.Module):
     def __init__(self):
         super().__init__()
@@ -84,9 +75,6 @@ def train_func_distributed():
         resources_per_worker={
             "CPU": 1,
         },
-        trainer_resources={
-            "CPU": 0,
-        },
     ),
 )
 

tests/e2e/support.py

@@ -122,9 +122,11 @@ def create_namespace_with_name(self, namespace_name):
         self.api_instance.create_namespace(namespace_body)
     except Exception as e:
         # Check if it's an AlreadyExists error (409 Conflict) and ignore it
-        if hasattr(e, 'status') and e.status == 409:
+        if hasattr(e, "status") and e.status == 409:
             # Namespace already exists, which is fine - just continue
-            print(f"Warning: Namespace '{namespace_name}' already exists, continuing...")
+            print(
+                f"Warning: Namespace '{namespace_name}' already exists, continuing..."
+            )
             return
         return _kube_api_error_handling(e)