Updates for 0.4.3 release

Author: Maxusmusti

demo-notebooks/batch-job/batch_mnist_ray.ipynb

@@ -324,6 +324,7 @@
    ],
    "source": [
     "jobdef = DDPJobDefinition(\n",
+    "    name=\"mnisttest\",\n",
     "    script=\"mnist.py\",\n",
     "    scheduler_args={\"requirements\": \"requirements.txt\"}\n",
     ")\n",

demo-notebooks/batch-job/mnist.py

@@ -14,7 +14,6 @@
 
 # In[]
 import os
-from tkinter import W
 
 import torch
 from pytorch_lightning import LightningModule, Trainer
@@ -150,7 +149,7 @@ def test_dataloader(self):
 trainer = Trainer(
     accelerator="auto",
     # devices=1 if torch.cuda.is_available() else None,  # limiting got iPython runs
-    max_epochs=100,
+    max_epochs=5,
     callbacks=[TQDMProgressBar(refresh_rate=20)],
     num_nodes=int(os.environ.get("GROUP_WORLD_SIZE", 1)),
     devices=int(os.environ.get("LOCAL_WORLD_SIZE", 1)),

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codeflare-sdk"
-version = "0.4.2"
+version = "0.4.3"
 description = "Python SDK for codeflare client"
 
 license = "Apache-2.0"