fix random_state typing (#18)

zenml-io · May 27, 2024 · a24daa8 · a24daa8
1 parent 81a90ba
commit a24daa8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/template/pipelines/inference.py b/template/pipelines/inference.py
@@ -12,7 +12,7 @@
 
 
 @pipeline
-def inference(random_state: str, target: str):
+def inference(random_state: int, target: str):
     """
     Model inference pipeline.