From fcb3abccf16418df2831da038b9736dc9016a058 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabrielf.almeida90@gmail.com>
Date: Tue, 14 Mar 2023 11:00:56 -0300
Subject: [PATCH] style: formatting and fixes

---
 Makefile                                             |  2 +-
 langflow/backend/langflow_backend/__init__.py        |  2 +-
 langflow/backend/langflow_backend/__main__.py        | 12 +++---------
 .../backend/langflow_backend/interface/signature.py  |  2 +-
 langflow/backend/langflow_backend/server.py          |  2 +-
 tests/test_loading.py                                |  2 +-
 6 files changed, 8 insertions(+), 14 deletions(-)

diff --git a/Makefile b/Makefile
index 314da2d99..e494f2a0d 100644
--- a/Makefile
+++ b/Makefile
@@ -18,7 +18,7 @@ format:
 lint:
 	poetry run mypy .
 	poetry run black . --check
-	poetry run ruff .
+	poetry run ruff . --fix
 
 install_frontend:
 	cd langflow/frontend && npm install
diff --git a/langflow/backend/langflow_backend/__init__.py b/langflow/backend/langflow_backend/__init__.py
index 1ece2c9f9..9285a0512 100644
--- a/langflow/backend/langflow_backend/__init__.py
+++ b/langflow/backend/langflow_backend/__init__.py
@@ -1 +1 @@
-from langflow_backend.interface.loading import load_flow_from_json
+from langflow_backend.interface.loading import load_flow_from_json  # noqa
diff --git a/langflow/backend/langflow_backend/__main__.py b/langflow/backend/langflow_backend/__main__.py
index ee3a5e7d5..544a5b8e5 100644
--- a/langflow/backend/langflow_backend/__main__.py
+++ b/langflow/backend/langflow_backend/__main__.py
@@ -1,5 +1,4 @@
 import multiprocessing
-import os
 import platform
 
 from langflow_backend.main import create_app
@@ -10,16 +9,14 @@ from pathlib import Path
 
 
 def get_number_of_workers(workers=None):
-    if workers is None:
-        workers = 1
-    elif workers == -1:
+    if workers == -1:
         workers = (multiprocessing.cpu_count() * 2) + 1
     return workers
 
 
 def serve(
-    workers: int = None,
-    timeout: int = None,
+    workers: int = 1,
+    timeout: int = 60,
 ):
     app = create_app()
     # get the directory of the current file
@@ -31,9 +28,6 @@ def serve(
         name="static",
     )
 
-    if not timeout:
-        timeout = 60
-
     host = "127.0.0.1"
     port = 5003
     options = {
diff --git a/langflow/backend/langflow_backend/interface/signature.py b/langflow/backend/langflow_backend/interface/signature.py
index c5d24d560..247c79789 100644
--- a/langflow/backend/langflow_backend/interface/signature.py
+++ b/langflow/backend/langflow_backend/interface/signature.py
@@ -120,7 +120,7 @@ def get_tool_signature(name: str):
         template["aiosession"]["required"] = False
         template["aiosession"]["show"] = False
 
-    template["_type"] = tool_type
+    template["_type"] = tool_type  # type: ignore
 
     return {
         "template": template,
diff --git a/langflow/backend/langflow_backend/server.py b/langflow/backend/langflow_backend/server.py
index 0e06bb9fb..a96f8b7e0 100644
--- a/langflow/backend/langflow_backend/server.py
+++ b/langflow/backend/langflow_backend/server.py
@@ -1,4 +1,4 @@
-from gunicorn.app.base import BaseApplication
+from gunicorn.app.base import BaseApplication  # type: ignore
 
 
 class LangflowApplication(BaseApplication):
diff --git a/tests/test_loading.py b/tests/test_loading.py
index 6ad3e72d9..3f0396f65 100644
--- a/tests/test_loading.py
+++ b/tests/test_loading.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from langflow import load_flow_from_json
+from langflow_backend import load_flow_from_json
 
 
 def test_load_flow_from_json():