fix(sdk): Fix datetime import (#1429)

kubeflow · Dec 18, 2023 · 066940a · 066940a
1 parent eaa2de6
commit 066940a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/kfp_tekton/_client.py b/sdk/python/kfp_tekton/_client.py
@@ -17,7 +17,7 @@
 import tempfile
 import time
 
-from datetime import datetime
+import datetime
 from typing import Mapping, Callable, Optional
 
 import kfp
@@ -832,7 +832,7 @@ def create_run_from_pipeline_func(self,
 
       # TODO: Check arguments against the pipeline function
       pipeline_name = pipeline_func.__name__
-      run_name = run_name or pipeline_name + ' ' + datetime.now().strftime('%Y-%m-%d %H-%M-%S')
+      run_name = run_name or pipeline_name + ' ' + datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S')
       try:
           (_, pipeline_package_path) = tempfile.mkstemp(suffix='.zip')
           TektonCompiler().compile(pipeline_func,