chore: pylint fixes

edx · Oct 23, 2023 · 7beb74f · 7beb74f
1 parent f0f0511
commit 7beb74f
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/flashcards/utils.py b/flashcards/utils.py
@@ -1,5 +1,10 @@
+"""
+Integration with openai to generate 'flashcards' in csv form
+based on course content
+"""
+
 import openai
-from flashcards.settings.private import OPENAI_API_KEY
+from flashcards.settings.private import OPENAI_API_KEY # pylint: disable=import-error,no-name-in-module
 
 openai.api_key = OPENAI_API_KEY
 
@@ -181,15 +186,15 @@
 
 
 c3 = openai.ChatCompletion.create(
-    model="gpt-3.5-turbo", 
+    model="gpt-3.5-turbo",
     messages=messages,
     temperature=1.0,
 )
 
 print(c3['choices'][0]['message']['content'])
 
 # c4 = openai.ChatCompletion.create(
-#     model="gpt-4", 
+#     model="gpt-4",
 #     messages=messages,
 #     temperature=1.0,
 # )