From a24daa8b34f4dc1ee8acaa537a5b511f2df411bb Mon Sep 17 00:00:00 2001 From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com> Date: Mon, 27 May 2024 20:49:34 +0200 Subject: [PATCH] fix random_state typing (#18) --- template/pipelines/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/template/pipelines/inference.py b/template/pipelines/inference.py index 691f58b..3449a2c 100644 --- 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.