Skip to content

Commit

Permalink
🔥 satisfy pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
nils-schmitt committed May 24, 2024
1 parent cae2237 commit 1c0c2ae
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 4 deletions.
4 changes: 3 additions & 1 deletion tracex_project/extraction/admin.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
"""Admin file for extraction app."""
from django.contrib import admin
from typing import Union

from django.contrib import admin

from extraction.models import Event, PatientJourney, Prompt, Trace, Cohort, Metric


Expand Down
3 changes: 2 additions & 1 deletion tracex_project/extraction/forms.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
"""Implementation of forms for the extraction app."""
from django import forms
from typing import List, Tuple

from django import forms

from extraction.models import PatientJourney
from tracex.forms import BaseEventForm
from tracex.logic.constants import MODULES_OPTIONAL
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@ def __extract_activities(patient_journey_numbered: str, condition: Optional[str]

user_message = patient_journey_numbered
if condition is not None:
user_message = f"Focus on those events that are related to the course of the disease of {condition}.\n\n{user_message}"
user_message = f"Focus on those events that are related to the course of the disease of {condition}.\n\n\
{user_message}"
messages.append({"role": "user", "content": user_message})
activity_labels = u.query_gpt(messages).split("\n")
df = pd.DataFrame(activity_labels, columns=[column_name])
Expand Down
2 changes: 1 addition & 1 deletion tracex_project/extraction/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,11 @@
"""

import traceback
from typing import Dict, List

# pylint: disable=unused-argument, unused-variable

import pandas as pd
from typing import Dict, List

from django.urls import reverse_lazy
from django.views import generic
Expand Down

0 comments on commit 1c0c2ae

Please sign in to comment.