[bug fixes] ensure step_init of parent decorator is called. (#6)

* changing command style to ensure that if alias is missing then it still works.

Author: valayDave

metaflow_extensions/torchrun/plugins/torchrun_libs/executor.py

@@ -69,7 +69,7 @@ def _exec_cmd(
         self._ensure_torch_installed()
 
         # Container to build up the command to be run in a subprocess.
-        cmd = ["torchrun"]
+        cmd = [sys.executable, "-m", "torch.distributed.run"]
 
         # Construct the torchrun distributed arguments.
 

metaflow_extensions/torchrun/plugins/torchrun_libs/torchrun_decorator.py

@@ -48,6 +48,7 @@ def _setup_current(self, main_addr, main_port, ubf_context, num_nodes, node_inde
         )
 
     def step_init(self, flow, graph, step, decos, environment, flow_datastore, logger):
+        super().step_init(flow, graph, step, decos, environment, flow_datastore, logger)
 
         self.flow_datastore = flow_datastore
 

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_namespace_packages
 
-version = "0.0.8"
+version = "0.0.9"
 
 setup(
     name="metaflow-torchrun",