Update my_solution.py

my_solution.py

@@ -35,7 +35,7 @@ def convert_entry_to_human_readable(entry):
 
 
 def predict(entry, visualize=False) -> Tuple[np.ndarray, List[int]]:
-    return entry['__key__'], np.zeros((21,3)), [(0, 1)]            
+    return entry['__key__'], np.zeros((17,3)), [(0, 1)]            
     try:
         solver = GeomSolver(entry)
         vertices, edges = solver.get_vertices()