style: lint

skit-ai · Aug 29, 2021 · 7545d5a · 7545d5a
1 parent f3fc4aa
commit 7545d5a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/dialogy/base/plugin.py b/dialogy/base/plugin.py
@@ -116,15 +116,19 @@ def __call__(self) -> PluginFn:
         """
         return self.plugin
 
-    def train(self, training_data: Any) -> Any:  # pylint: disable=unused-argument disable=no-self-use
+    def train(
+        self, training_data: Any
+    ) -> Any:  # pylint: disable=unused-argument disable=no-self-use
         """
         Train a plugin.
 
         This method is called by the workflow after the pre-processing stage.
         """
         return None
 
-    def transform(self, training_data: Any) -> Any:  # pylint: disable=unused-argument disable=no-self-use
+    def transform(
+        self, training_data: Any
+    ) -> Any:  # pylint: disable=unused-argument disable=no-self-use
         """
         Transform data for a plugin in the workflow.
         """

diff --git a/dialogy/plugins/text/classification/xlmr.py b/dialogy/plugins/text/classification/xlmr.py
@@ -218,4 +218,4 @@ def load(self) -> None:
         )
 
     def utility(self, *args: Any) -> Any:
-        return self.inference(*args) # pylint: disable=no-value-for-parameter
+        return self.inference(*args)  # pylint: disable=no-value-for-parameter