From ae2e993680b9544420a89ca73d52bb422c81e9f8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 09:34:45 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(manager.py):=20fix=20import?= =?UTF-8?q?=20statement=20to=20include=20Dict=20in=20typing=20module=20?= =?UTF-8?q?=E2=9C=A8=20feat(manager.py):=20refactor=20check=5Ftable=20meth?= =?UTF-8?q?od=20in=20DatabaseManager=20class=20to=20use=20list=20comprehen?= =?UTF-8?q?sion=20for=20improved=20readability=20and=20performance?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/database/manager.py | 10 ++++------ src/backend/langflow/services/manager.py | 2 +- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/src/backend/langflow/services/database/manager.py b/src/backend/langflow/services/database/manager.py index 7f8afab6f..a89db8395 100644 --- a/src/backend/langflow/services/database/manager.py +++ b/src/backend/langflow/services/database/manager.py @@ -107,12 +107,10 @@ class DatabaseManager(Service): # We will check that all models are in the database # and that the database is up to date with all columns sql_models = [models.Flow, models.User, models.ApiKey] - results = [] - for sql_model in sql_models: - results.append( - TableResults(sql_model.__tablename__, self.check_table(sql_model)) - ) - return results + return [ + TableResults(sql_model.__tablename__, self.check_table(sql_model)) + for sql_model in sql_models + ] def check_table(self, model): results = [] diff --git a/src/backend/langflow/services/manager.py b/src/backend/langflow/services/manager.py index 34c58c648..dc901730c 100644 --- a/src/backend/langflow/services/manager.py +++ b/src/backend/langflow/services/manager.py @@ -1,5 +1,5 @@ from langflow.services.schema import ServiceType -from typing import TYPE_CHECKING, List, Optional +from typing import TYPE_CHECKING, Dict, List, Optional from loguru import logger if TYPE_CHECKING: