Skip to content

Commit

Permalink
Reformat with black
Browse files Browse the repository at this point in the history
  • Loading branch information
OlivieFranklova committed Sep 18, 2024
1 parent 42e0fb9 commit b88e01d
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 2 deletions.
1 change: 0 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,6 @@ scipy==1.14.1
sentence-transformers==3.1.0
six==1.16.0
smmap==5.0.1
streamlit==1.38.0
sympy==1.13.2
tenacity==9.0.0
threadpoolctl==3.5.0
Expand Down
1 change: 1 addition & 0 deletions similarityRunner/interfaces/ConnectorInterface.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
File contains Connector interface
"""

import abc

from models.connector_models import ConnectorSettings, Output, ConnectorOutput
Expand Down
1 change: 1 addition & 0 deletions similarityRunner/interfaces/UserInterface.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
File contains UserInterface interface
"""

import abc

from models.user_models import SimilarityOutput
Expand Down
3 changes: 3 additions & 0 deletions similarityRunner/models/connector_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
- the base class for connector settings and derived classes.
- the base class for connector output and derived classes.
"""

import pandas as pd
from pydantic import BaseModel

Expand All @@ -13,11 +14,13 @@ class ConnectorSettings(BaseModel):
"""
ConnectorSettings class is a base class for connector settings.
"""

# here will be common fields for all connectors


class ConnectorOutput(BaseModel):
"""
ConnectorOutput class is a base class for connector output.
"""

# here will be common fields for all connectors
4 changes: 3 additions & 1 deletion similarityRunner/models/user_models.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,15 @@
"""
This module contains the user models
"""

from pydantic import BaseModel


class SimilarityOutput(BaseModel):
"""
SimilarityOutput class isclass containing similarity output.
"""

# here will be common fields for all similarity models
table_names: list[str]
distances: dict[(str, str),float]
distances: dict[(str, str), float]

0 comments on commit b88e01d

Please sign in to comment.