Update main.py

main.py

@@ -1,7 +1,7 @@
 import os
 import sys
 import math
-from typing import Optional, Dict, Any, list
+from typing import Optional, Dict, Any, List
 
 import numpy as np
 from numpy.linalg import norm