From 5b72b4e0078d45d7729a029789af906c762a5073 Mon Sep 17 00:00:00 2001 From: kalvdans Date: Wed, 5 Apr 2017 06:22:33 +0200 Subject: [PATCH] Avoid warnings in unit tests. (#2078) --- test/worker_external_task_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/worker_external_task_test.py b/test/worker_external_task_test.py index ae0e71ffa8..d2030c63fe 100644 --- a/test/worker_external_task_test.py +++ b/test/worker_external_task_test.py @@ -13,7 +13,7 @@ # the License. import luigi -from luigi.file import LocalTarget +from luigi.local_target import LocalTarget from luigi.scheduler import Scheduler import luigi.server import luigi.worker @@ -28,7 +28,7 @@ class TestExternalFileTask(luigi.ExternalTask): """ Mocking tasks is a pain, so touch a file instead """ path = luigi.Parameter() - times_to_call = luigi.Parameter() + times_to_call = luigi.IntParameter() def __init__(self, *args, **kwargs): super(TestExternalFileTask, self).__init__(*args, **kwargs) @@ -54,7 +54,7 @@ class TestTask(luigi.Task): Requires a single file dependency """ tempdir = luigi.Parameter() - complete_after = luigi.Parameter() + complete_after = luigi.IntParameter() def __init__(self, *args, **kwargs): super(TestTask, self).__init__(*args, **kwargs)