Fix hello-pt-cse job (#2905)

NVIDIA · Sep 4, 2024 · dc6acfd · dc6acfd
1 parent 60e722a
commit dc6acfd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/integration_test/data/jobs/hello-pt-cse/app/custom/train.py b/tests/integration_test/data/jobs/hello-pt-cse/app/custom/train.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import argparse
+import os
 
 import torch
 import torch.nn as nn