Update api/gpu_manager.py

api/gpu_manager.py

@@ -8,6 +8,7 @@ import os
 import torch
 import math
 import logging
+from typing import List
 
 class GPUManager:
     """