aiagents4pharma 1.43.0__py3-none-any.whl → 1.45.0__py3-none-any.whl

aiagents4pharma/talk2cells/tools/__init__.py

@@ -1,5 +1,6 @@
-'''
+"""
 This file is used to import all the modules in the package.
-'''
+"""
+
 # import everything from the module
 from . import scp_agent

aiagents4pharma/talk2cells/tools/scp_agent/__init__.py

@@ -1,6 +1,6 @@
-'''
+"""
 This file is used to import all the modules in the package.
-'''
+"""
+
 # import everything from the module
-from . import display_studies
-from . import search_studies
+from . import display_studies, search_studies

aiagents4pharma/talk2cells/tools/scp_agent/display_studies.py

@@ -1,11 +1,12 @@
 #!/usr/bin/env python3
 
-'''
+"""
 This tool is used to display the table of studies.
-'''
+"""
 
 import logging
 from typing import Annotated
+
 from langchain_core.tools import tool
 from langgraph.prebuilt import InjectedState
 
@@ -13,7 +14,8 @@ from langgraph.prebuilt import InjectedState
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-@tool('display_studies')
+
+@tool("display_studies")
 def display_studies(state: Annotated[dict, InjectedState]):
     """
     Display the table of studies.

aiagents4pharma/talk2cells/tools/scp_agent/search_studies.py

@@ -1,26 +1,28 @@
 #!/usr/bin/env python3
 
-'''
+"""
 A tool to fetch studies from the Single Cell Portal.
-'''
+"""
 
 import logging
 from typing import Annotated
+
+import pandas as pd
 import requests
+from langchain_core.messages import ToolMessage
 from langchain_core.tools import tool
 from langchain_core.tools.base import InjectedToolCallId
-from langchain_core.messages import ToolMessage
 from langgraph.types import Command
-import pandas as pd
 
 # Initialize logger
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-@tool('search_studies')
-def search_studies(search_term: str,
-                   tool_call_id: Annotated[str, InjectedToolCallId],
-                   limit: int = 5):
+
+@tool("search_studies")
+def search_studies(
+    search_term: str, tool_call_id: Annotated[str, InjectedToolCallId], limit: int = 5
+):
     """
     Fetch studies from single cell portal
 
@@ -30,16 +32,13 @@ def search_studies(search_term: str,
 
     """
     logger.log(logging.INFO, "Calling the tool search_studies")
-    scp_endpoint = 'https://singlecell.broadinstitute.org/single_cell/api/v1/search?type=study'
+    scp_endpoint = "https://singlecell.broadinstitute.org/single_cell/api/v1/search?type=study"
     # params = {'terms': search_term, 'facets': 'MONDO_0005011'}
-    params = {'terms': search_term}
+    params = {"terms": search_term}
     status_code = 0
     while status_code != 200:
         # Make a GET request to the single cell portal
-        search_response = requests.get(scp_endpoint,
-                                       params=params,
-                                       timeout=10,
-                                       verify=False)
+        search_response = requests.get(scp_endpoint, params=params, timeout=10, verify=False)
         status_code = search_response.status_code
         logger.log(logging.INFO, "Status code %s received from SCP")
 
@@ -48,20 +47,20 @@ def search_studies(search_term: str,
 
     # Extract the data from the response
     # with the selected columns
-    df = pd.DataFrame(search_response.json()['studies'])[selected_columns]
+    df = pd.DataFrame(search_response.json()["studies"])[selected_columns]
 
     # Convert column 'Study Name' into clickable
     # hyperlinks from the column 'Study URL'
-    scp_api_url = 'https://singlecell.broadinstitute.org'
-    df['name'] = df.apply(
-            lambda x: f"<a href=\"{scp_api_url}/{x['study_url']}\">{x['name']}</a>",
-            axis=1)
+    scp_api_url = "https://singlecell.broadinstitute.org"
+    df["name"] = df.apply(
+        lambda x: f'<a href="{scp_api_url}/{x["study_url"]}">{x["name"]}</a>', axis=1
+    )
 
     # Excldue the column 'Study URL' from the dataframe
-    df = df.drop(columns=['study_url'])
+    df = df.drop(columns=["study_url"])
 
     # Add a new column a the beginning of the dataframe with row numbers
-    df.insert(0, 'S/N', range(1, 1 + len(df)))
+    df.insert(0, "S/N", range(1, 1 + len(df)))
 
     # Update the state key 'search_table' with the dataframe in markdown format
     return Command(
@@ -72,7 +71,7 @@ def search_studies(search_term: str,
             "messages": [
                 ToolMessage(
                     f"Successfully fetched {limit} studies on {search_term}.",
-                    tool_call_id=tool_call_id
+                    tool_call_id=tool_call_id,
                 )
             ],
         }

aiagents4pharma/talk2knowledgegraphs/.dockerignore

@@ -0,0 +1,13 @@
+_pycache_/
+*.pyc
+*.log
+*.csv
+*.pt
+*.pkl
+models/
+data/
+env/
+.venv/
+.git/
+.env
+.cufile.log

aiagents4pharma/talk2knowledgegraphs/Dockerfile

@@ -0,0 +1,103 @@
+# syntax=docker/dockerfile:1
+
+# Dockerfile for the talk2knowledgegraphs application
+# Multi-stage build for optimized image size with UV package manager
+
+ARG BASE_IMAGE=ubuntu:24.04
+ARG PYTHON_VERSION=3.12
+
+FROM ${BASE_IMAGE} AS dev-base
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  build-essential \
+  ca-certificates \
+  cmake \
+  curl \
+  g++ \
+  libopenblas-dev \
+  libomp-dev \
+  ninja-build \
+  wget \
+  && rm -rf /var/lib/apt/lists/*
+
+FROM dev-base AS python-install
+ARG PYTHON_VERSION=3.12
+
+# Install Python (available in Ubuntu 24.04 default repos)
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  python${PYTHON_VERSION} \
+  python${PYTHON_VERSION}-dev \
+  python${PYTHON_VERSION}-venv \
+  python3-pip \
+  && rm -rf /var/lib/apt/lists/* \
+  && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+  && update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
+
+FROM python-install AS uv-install
+WORKDIR /app
+
+# Install UV package manager and dependencies
+COPY pyproject.toml uv.lock* ./
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
+  export PATH="/root/.local/bin:$PATH" && \
+  export UV_PROJECT_ENVIRONMENT="/opt/venv" && \
+  uv sync --frozen --extra dev --no-install-project --python python${PYTHON_VERSION} && \
+  . /opt/venv/bin/activate && \
+  uv pip install \
+  --extra-index-url=https://pypi.nvidia.com \
+  --index-strategy unsafe-best-match \
+  cudf-cu12 dask-cudf-cu12 && \
+  uv cache clean
+
+FROM ${BASE_IMAGE} AS runtime
+ARG PYTHON_VERSION=3.12
+LABEL maintainer="talk2knowledgegraphs"
+LABEL version="1.0.0"
+LABEL description="AI Agents for Pharma - Knowledge Graphs Application"
+
+# Install runtime dependencies
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  ca-certificates \
+  curl \
+  libmagic1 \
+  libopenblas0 \
+  libomp5 \
+  python${PYTHON_VERSION} \
+  && rm -rf /var/lib/apt/lists/* \
+  && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+  && update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
+
+# Copy UV virtual environment from build stage
+COPY --from=uv-install /opt/venv /opt/venv
+
+# Set environment variables
+ENV PATH="/opt/venv/bin:$PATH"
+ENV PYTHONPATH="/app"
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV STREAMLIT_SERVER_HEADLESS=true
+ENV STREAMLIT_SERVER_ENABLE_CORS=false
+
+# Set working directory and create necessary directories
+WORKDIR /app
+
+# Copy application code
+COPY aiagents4pharma/talk2knowledgegraphs /app/aiagents4pharma/talk2knowledgegraphs
+COPY docs /app/docs
+COPY app /app/app
+
+# Copy and set up the entrypoint script
+COPY aiagents4pharma/talk2knowledgegraphs/entrypoint.sh /usr/local/bin/entrypoint.sh
+RUN chmod +x /usr/local/bin/entrypoint.sh
+
+# Health check for production monitoring
+HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
+  CMD curl -f http://localhost:8501/health || exit 1
+
+# Expose the default Streamlit port
+EXPOSE 8501
+
+# Set the entrypoint
+ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]
+
+# Default command (can be overridden)
+CMD ["streamlit", "run", "/app/app/frontend/streamlit_app_talk2knowledgegraphs.py", "--server.port=8501", "--server.address=0.0.0.0"]

aiagents4pharma/talk2knowledgegraphs/README.md

@@ -0,0 +1 @@
+Please check out the README file in the root folder for more information.

aiagents4pharma/talk2knowledgegraphs/__init__.py

@@ -1,8 +1,5 @@
-'''
+"""
 This file is used to import the datasets and utils.
-'''
-from . import agents
-from . import datasets
-from . import states
-from . import tools
-from . import utils
+"""
+
+from . import agents, datasets, states, tools, utils

aiagents4pharma/talk2knowledgegraphs/agents/__init__.py

@@ -1,4 +1,5 @@
-'''
+"""
 This file is used to import all the models in the package.
-'''
+"""
+
 from . import t2kg_agent

aiagents4pharma/talk2knowledgegraphs/agents/t2kg_agent.py

@@ -1,34 +1,42 @@
-'''
+"""
 This is the agent file for the Talk2KnowledgeGraphs agent.
-'''
+"""
 
 import logging
 from typing import Annotated
+
 import hydra
 from langchain_core.language_models.chat_models import BaseChatModel
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph import START, StateGraph
-from langgraph.prebuilt import create_react_agent, ToolNode, InjectedState
+from langgraph.prebuilt import InjectedState, ToolNode, create_react_agent
+
+from ..states.state_talk2knowledgegraphs import Talk2KnowledgeGraphs
+from ..tools.graphrag_reasoning import GraphRAGReasoningTool
+
 # from ..tools.multimodal_subgraph_extraction import MultimodalSubgraphExtractionTool
-from ..tools.milvus_multimodal_subgraph_extraction import MultimodalSubgraphExtractionTool
+from ..tools.milvus_multimodal_subgraph_extraction import (
+    MultimodalSubgraphExtractionTool,
+)
+
 # from ..tools.cu2_multimodal_subgraph_extraction import MultimodalSubgraphExtractionTool
 # from ..tools.gsfs_multimodal_subgraph_extraction import MultimodalSubgraphExtractionTool
 from ..tools.subgraph_summarization import SubgraphSummarizationTool
-from ..tools.graphrag_reasoning import GraphRAGReasoningTool
-from ..states.state_talk2knowledgegraphs import Talk2KnowledgeGraphs
 
 # Initialize logger
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+
 def get_app(uniq_id, llm_model: BaseChatModel):
-    '''
+    """
     This function returns the langraph app.
-    '''
+    """
+
     def agent_t2kg_node(state: Annotated[dict, InjectedState]):
-        '''
+        """
         This function calls the model.
-        '''
+        """
         logger.log(logging.INFO, "Calling t2kg_agent node with thread_id %s", uniq_id)
         response = model.invoke(state, {"configurable": {"thread_id": uniq_id}})
 
@@ -37,29 +45,30 @@ def get_app(uniq_id, llm_model: BaseChatModel):
     # Load hydra configuration
     logger.log(logging.INFO, "Load Hydra configuration for Talk2KnowledgeGraphs agent.")
     with hydra.initialize(version_base=None, config_path="../configs"):
-        cfg = hydra.compose(config_name='config',
-                            overrides=['agents/t2kg_agent=default'])
+        cfg = hydra.compose(config_name="config", overrides=["agents/t2kg_agent=default"])
         cfg = cfg.agents.t2kg_agent
 
     # Define the tools
     subgraph_extraction = MultimodalSubgraphExtractionTool()
     subgraph_summarization = SubgraphSummarizationTool()
     graphrag_reasoning = GraphRAGReasoningTool()
-    tools = ToolNode([
-                    subgraph_extraction,
-                    subgraph_summarization,
-                    graphrag_reasoning,
-                    ])
+    tools = ToolNode(
+        [
+            subgraph_extraction,
+            subgraph_summarization,
+            graphrag_reasoning,
+        ]
+    )
 
     # Create the agent
     model = create_react_agent(
-                llm_model,
-                tools=tools,
-                state_schema=Talk2KnowledgeGraphs,
-                prompt=cfg.state_modifier,
-                version='v2',
-                checkpointer=MemorySaver()
-            )
+        llm_model,
+        tools=tools,
+        state_schema=Talk2KnowledgeGraphs,
+        prompt=cfg.state_modifier,
+        version="v2",
+        checkpointer=MemorySaver(),
+    )
 
     # Define a new graph
     workflow = StateGraph(Talk2KnowledgeGraphs)
@@ -79,11 +88,12 @@ def get_app(uniq_id, llm_model: BaseChatModel):
     # meaning you can use it as you would any other runnable.
     # Note that we're (optionally) passing the memory
     # when compiling the graph
-    app = workflow.compile(checkpointer=checkpointer,
-                           name="T2KG_Agent")
-    logger.log(logging.INFO,
-               "Compiled the graph with thread_id %s and llm_model %s",
-               uniq_id,
-               llm_model)
+    app = workflow.compile(checkpointer=checkpointer, name="T2KG_Agent")
+    logger.log(
+        logging.INFO,
+        "Compiled the graph with thread_id %s and llm_model %s",
+        uniq_id,
+        llm_model,
+    )
 
     return app

aiagents4pharma/talk2knowledgegraphs/configs/__init__.py

@@ -1,8 +1,5 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""
 
-from . import agents
-from . import tools
-from . import app
-from . import utils
+from . import agents, app, tools, utils

aiagents4pharma/talk2knowledgegraphs/configs/agents/t2kg_agent/__init__.py

@@ -1,3 +1,3 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""

aiagents4pharma/talk2knowledgegraphs/configs/agents/t2kg_agent/default.yaml

@@ -1,7 +1,7 @@
 _target_: agents.t2kg_agent.get_app
 state_modifier: >
   You are talk2knowledgegraphs agent, a helpful assistant for reasoning over knowledge graphs.
-  User can ask questions related to the knowledge graphs, and you will provide the answers using 
+  User can ask questions related to the knowledge graphs, and you will provide the answers using
   the provided tools as follows (if necessary):
   [`subgraph_extraction`, `subgraph_summarization`, `graphrag_reasoning`].
 
@@ -17,19 +17,19 @@ state_modifier: >
   - `graphrag_reasoning`: Reason over the extracted textualized subgraph to answer the user's
     prompt by also considering the context from the extracted subgraph and the retrieved
     documents. User may also have a set of uploaded files that can be used to provide additional
-    information for reasoning. The history of previous conversations should be considered as well, 
+    information for reasoning. The history of previous conversations should be considered as well,
     and you as an agent should provide which conversations can be included as chat history.
 
-  As an agent, you should approach each request by first understanding the user's query and then 
-  following the appropriate steps to provide the best answer possible. 
+  As an agent, you should approach each request by first understanding the user's query and then
+  following the appropriate steps to provide the best answer possible.
 
   **Execution Steps**:
   - Understand thoroughly the user's query and think over the best approach to answer it.
-  - You may not need to call any tool for each user's query. Use the related tool(s) as needed. 
+  - You may not need to call any tool for each user's query. Use the related tool(s) as needed.
     Think deeply whether it is necessary to call any tool to respond to the user's request.
   - Call `subgraph_extraction` if there is any indication that the user needs to get the
     information from the knowledge graph, which is not directly available as context in the prompt or
-    in the previous extracted subgraph. 
+    in the previous extracted subgraph.
     If the user asks for subgraph extraction, suggest a value for the `extraction_name` argument.
     You should always follow it with `subgraph_summarization` as the next tool to be invoked.
   - Call `subgraph_summarization` tool to summarize the extracted subgraph and provide
@@ -43,7 +43,7 @@ state_modifier: >
     the answer can be retrieved from the previously extracted subgraph.
   - Call `graphrag_reasoning` tool to reason over the extracted subgraph and documents.
     Always perform reasoning over the extracted subgraph and documents to provide
-    the best possible answer to the user's query. Before calling this tool, 
+    the best possible answer to the user's query. Before calling this tool,
     make sure you have access to the summarized subgraph obtained from `subgraph_summarization` tool.
   - By default, if the user asks for a specific question about the extracted graph, you should
     call `subgraph_summarization` followed by `graphrag_reasoning` if the most recent subgraphs
@@ -59,4 +59,4 @@ state_modifier: >
   is not enough to answer the user's question.
   - `subgraph_summarization` -> `graphrag_reasoning` when the user asks for the previously extracted
   subgraph. Use the summarized subgraph as the subgraph context in the `graphrag_reasoning` tool.
-  - Do not call `graphrag_reasoning` tool without calling `subgraph_summarization` tool first.
+  - Do not call `graphrag_reasoning` tool without calling `subgraph_summarization` tool first.

aiagents4pharma/talk2knowledgegraphs/configs/app/__init__.py

@@ -1,4 +1,5 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""
+
 from . import frontend

aiagents4pharma/talk2knowledgegraphs/configs/app/frontend/__init__.py

@@ -1,3 +1,3 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""

aiagents4pharma/talk2knowledgegraphs/configs/app/frontend/default.yaml

@@ -60,4 +60,4 @@ milvus_db:
   collection_nodes_drug: "t2kg_primekg_nodes_drug"
   collection_nodes_disease: "t2kg_primekg_nodes_disease"
   query_batch_size: 1000000
-  cache_edge_index_path: "${oc.env:CACHE_EDGE_INDEX_PATH,aiagents4pharma/talk2knowledgegraphs/tests/files/t2kg_primekg_edge_index.pkl}"
+  cache_edge_index_path: "${oc.env:CACHE_EDGE_INDEX_PATH,aiagents4pharma/talk2knowledgegraphs/tests/files/t2kg_primekg_edge_index.pkl}"

aiagents4pharma/talk2knowledgegraphs/configs/config.yaml

@@ -9,4 +9,4 @@ defaults:
   - utils/enrichments/uniprot_proteins: default
   - utils/enrichments/ols_terms: default
   - utils/enrichments/reactome_pathways: default
-  - app/frontend: default
+  - app/frontend: default

aiagents4pharma/talk2knowledgegraphs/configs/tools/__init__.py

@@ -1,6 +1,5 @@
-'''
+"""
 Import all the modules in the package
-'''
-from . import subgraph_extraction
-from . import subgraph_summarization
-from . import graphrag_reasoning
+"""
+
+from . import graphrag_reasoning, subgraph_extraction, subgraph_summarization

aiagents4pharma/talk2knowledgegraphs/configs/tools/graphrag_reasoning/__init__.py

@@ -1,3 +1,3 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""

aiagents4pharma/talk2knowledgegraphs/configs/tools/graphrag_reasoning/default.yaml

@@ -21,4 +21,4 @@ prompt_graphrag_w_docs: >
 
   Subgraph Summary: {subgraph_summary}
   Context: {context}
-  Question: {input}
+  Question: {input}

aiagents4pharma/talk2knowledgegraphs/configs/tools/multimodal_subgraph_extraction/default.yaml

@@ -3,6 +3,8 @@ ollama_embeddings:
   - "nomic-embed-text"
 temperature: 0.1
 streaming: False
+
+# PCST Algorithm Parameters
 topk: 15
 topk_e: 15
 cost_e: 0.5
@@ -11,7 +13,20 @@ root: -1
 num_clusters: 1
 pruning: "gw"
 verbosity_level: 0
-search_metric_type: "IP"  # Inner Product
+
+# Hardware-Specific Vector Processing
+vector_processing:
+  # Enable dynamic metric type selection based on hardware
+  dynamic_metrics: true
+
+# Milvus Database Configuration
+milvus_db:
+  # Database and collection names
+  database_name: "t2kg_primekg"
+
+  # Cache settings for edge index
+  cache_edge_index_path: "${oc.env:CACHE_EDGE_INDEX_PATH,aiagents4pharma/talk2knowledgegraphs/tests/files/t2kg_primekg_edge_index.pkl}"
+
 node_id_column: "node_id"
 node_attr_column: "node_attr"
 edge_src_column: "edge_src"
@@ -34,4 +49,4 @@ biobridge:
     - "cellular_component"
     - "biological_process"
     - "drug"
-    - "disease"
+    - "disease"

aiagents4pharma/talk2knowledgegraphs/configs/tools/subgraph_extraction/__init__.py

@@ -1,3 +1,3 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""

aiagents4pharma/talk2knowledgegraphs/configs/tools/subgraph_extraction/default.yaml

@@ -40,4 +40,4 @@ splitter_chunk_overlap: 16
 retriever_search_type: "mmr"
 retriever_k: 3
 retriever_fetch_k: 10
-retriever_lambda_mult: 0.3
+retriever_lambda_mult: 0.3

aiagents4pharma/talk2knowledgegraphs/configs/tools/subgraph_summarization/__init__.py

@@ -1,3 +1,3 @@
-'''
+"""
 Import all the modules in the package
-'''
+"""

aiagents4pharma/talk2knowledgegraphs/configs/tools/subgraph_summarization/default.yaml

@@ -6,4 +6,4 @@ prompt_subgraph_summarization: >
   highlighting the most important nodes and edges in the subgraph to respond to the user's question.
 
   Textualized Subgraph: {textualized_subgraph}
-  Question: {input}
+  Question: {input}

aiagents4pharma/talk2knowledgegraphs/configs/utils/enrichments/ols_terms/default.yaml

@@ -1,3 +1,3 @@
 _target_: talk2knowledgegraphs.utils.ols_terms
 base_url: https://www.ebi.ac.uk/ols4/api/terms/
-timeout: 10
+timeout: 10

aiagents4pharma/talk2knowledgegraphs/configs/utils/enrichments/reactome_pathways/default.yaml

@@ -1,3 +1,3 @@
 _target_: talk2knowledgegraphs.utils.reactome_pathways
 base_url: https://reactome.org/ContentService/data/query/
-timeout: 10
+timeout: 10

aiagents4pharma/talk2knowledgegraphs/configs/utils/enrichments/uniprot_proteins/default.yaml

@@ -3,4 +3,4 @@ uniprot_url: https://www.ebi.ac.uk/proteins/api/proteins
 reviewed: true
 isoform: 0
 organism: Homo sapiens
-timeout: 10
+timeout: 10

aiagents4pharma/talk2knowledgegraphs/configs/utils/pubchem_utils/default.yaml

@@ -2,4 +2,4 @@ _target_: talk2knowledgegraphs.utils.pubchem_utils
 pubchem_cid_base_url: https://pubchem.ncbi.nlm.nih.gov/rest/pug/substance/sourceid/
 pubchem_cid_description_url: https://pubchem.ncbi.nlm.nih.gov/rest/pug/compound/cid/
 pubchem_casrn2cid_url: https://pubchem.ncbi.nlm.nih.gov/rest/pug/substance/xref/RN/
-pubchem_cid2smiles_url: https://pubchem.ncbi.nlm.nih.gov/rest/pug/compound/cid/
+pubchem_cid2smiles_url: https://pubchem.ncbi.nlm.nih.gov/rest/pug/compound/cid/

aiagents4pharma/talk2knowledgegraphs/datasets/__init__.py

@@ -1,7 +1,5 @@
-'''
+"""
 This file is used to import all the models in the package.
-'''
-from . import dataset
-from . import primekg
-from . import starkqa_primekg
-from . import biobridge_primekg
+"""
+
+from . import biobridge_primekg, dataset, primekg, starkqa_primekg