Flake8

spotify · Oct 23, 2021 · e29a937 · e29a937
1 parent 59ec27d
commit e29a937
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/luigi/__init__.py b/luigi/__init__.py
@@ -63,7 +63,7 @@
     'FloatParameter', 'BoolParameter', 'TaskParameter',
     'ListParameter', 'TupleParameter', 'EnumParameter', 'DictParameter',
     'configuration', 'interface', 'local_target', 'run', 'build', 'event', 'Event',
-    'NumericalParameter', 'ChoiceParameter', 'OptionalParameter',
+    'NumericalParameter', 'ChoiceParameter', 'OptionalParameter', 'OptionalStrParameter',
     'OptionalIntParameter', 'OptionalFloatParameter', 'OptionalBoolParameter',
     'OptionalDictParameter', 'OptionalListParameter', 'OptionalTupleParameter',
     'OptionalChoiceParameter', 'OptionalNumericalParameter', 'LuigiStatusCode',

diff --git a/test/parameter_test.py b/test/parameter_test.py
@@ -387,7 +387,10 @@ class TestConfig(luigi.Config):
             param = luigi.OptionalParameter()
 
         TestConfig(param=1)
-        warnings.warn.assert_called_once_with('OptionalParameter "param" with value "1" is not of type "str" or None.', luigi.parameter.OptionalParameterTypeWarning)
+        warnings.warn.assert_called_once_with(
+            'OptionalParameter "param" with value "1" is not of type "str" or None.',
+            luigi.parameter.OptionalParameterTypeWarning
+        )
 
     def test_optional_parameter_parse_none(self):
         self.assertIsNone(luigi.OptionalParameter().parse(''))