Make linters happy

Author: PawelPeczek-Roboflow

roboflow/adapters/rfapi.py

@@ -1,7 +1,7 @@
 import json
 import os
 import urllib
-from typing import List, Optional
+from typing import List, Optional, Dict, Union
 
 import requests
 from requests.exceptions import RequestException
@@ -67,7 +67,7 @@ def start_version_training(
     """
     url = f"{API_URL}/{workspace_url}/{project_url}/{version}/train?api_key={api_key}&nocache=true"
 
-    data = {}
+    data: Dict[str, Union[str, int]] = {}
     if speed is not None:
         data["speed"] = speed
     if checkpoint is not None:

roboflow/core/version.py

@@ -307,7 +307,7 @@ def train(
             model_type: The type of model to train. Default depends on kind of project. It takes precedence over speed. You can check the list of model ids by sending an invalid parameter in this argument.
             checkpoint: A string representing the checkpoint to use while training
             epochs: Number of epochs to train the model
-            plot: Whether to plot the training results. Default is `False`.
+            plot_in_notebook: Whether to plot the training results. Default is `False`.
 
         Returns:
             An instance of the trained model class
@@ -389,15 +389,15 @@ def live_plot(epochs, mAP, loss, title=""):
                         write_line(line="Training failed")
                         break
 
-            epochs: Union[np.ndarray, list]
+            epoch_ids: Union[np.ndarray, list]
             mAP: Union[np.ndarray, list]
             loss: Union[np.ndarray, list]
 
             if "roboflow-train" in models.keys():
                 import numpy as np
 
                 # training has started
-                epochs = np.array([int(epoch["epoch"]) for epoch in models["roboflow-train"]["epochs"]])
+                epoch_ids = np.array([int(epoch["epoch"]) for epoch in models["roboflow-train"]["epochs"]])
                 mAP = np.array([float(epoch["mAP"]) for epoch in models["roboflow-train"]["epochs"]])
                 loss = np.array(
                     [
@@ -414,23 +414,23 @@ def live_plot(epochs, mAP, loss, title=""):
                     num_machine_spin_dots = ["."]
                 title = "Training Machine Spinning Up" + "".join(num_machine_spin_dots)
 
-                epochs = []
+                epoch_ids = []
                 mAP = []
                 loss = []
 
-            if (len(epochs) > len(previous_epochs)) or (len(epochs) == 0):
+            if (len(epoch_ids) > len(previous_epochs)) or (len(epoch_ids) == 0):
                 if plot_in_notebook:
-                    live_plot(epochs, mAP, loss, title)
+                    live_plot(epoch_ids, mAP, loss, title)
                 else:
-                    if len(epochs) > 0:
+                    if len(epoch_ids) > 0:
                         title = (
-                            title + ": Epoch: " + str(epochs[-1]) + " mAP: " + str(mAP[-1]) + " loss: " + str(loss[-1])
+                                title + ": Epoch: " + str(epoch_ids[-1]) + " mAP: " + str(mAP[-1]) + " loss: " + str(loss[-1])
                         )
                     if not first_graph_write:
                         write_line(title)
                         first_graph_write = True
 
-            previous_epochs = copy.deepcopy(epochs)
+            previous_epochs = copy.deepcopy(epoch_ids)
 
             time.sleep(5)