fix

.idea/workspace.xml

@@ -61,7 +61,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1713919907805</updated>
-      <workItem from="1713919910236" duration="394000" />
+      <workItem from="1713919910236" duration="599000" />
     </task>
     <servers />
   </component>

Kaggle_API.py

@@ -9,7 +9,7 @@ class API_Connection:
         os.environ["KAGGLE_USERNAME"] = kaggle_username
         os.environ["KAGGLE_KEY"] = kaggle_key
 
-        self.GoogleDrive_connection = gd_connection
+        self.GoogleDrive_connection: GoogleDrive_API = gd_connection
 
         self.NOTEBOOK_ID = "amirmorris/pix2pix-model"
         self.DATASET_NAME = "dataset"