for now, fix to train until we fix naming

.github/workflows/runner.py

@@ -14,15 +14,15 @@
 if config["lang"] == "cu":
     comp, run = run_cuda_script(
         {"eval.cu": cu_eval},
-        {key: config[key] for key in ["reference.cuh", "submission.cuh"] if key in config},
+        {key: config[key] for key in ["reference.cuh", "train.cuh"] if key in config},
         arch=None,
     )
     result = {"compile": asdict(comp), "run": asdict(run)}
 else:
     run = run_pytorch_script(
         {
             "eval.py": py_eval,
-            **{key: config[key] for key in ["reference.py", "submission.py"] if key in config},
+            **{key: config[key] for key in ["reference.py", "train.py"] if key in config},
         },
         main="eval.py",
         arch=None,