Hugging Face's logo

rozumden
/
DRAK

Model card Files
xet
Community
Denys Rozumnyi commited on
Commit
a99680e
·
1 Parent(s): 8439fac

update

Browse files
Files changed (2) hide show
  1. __pycache__/my_solution.cpython-39.pyc +0 -0
  2. my_solution.py +2 -0
__pycache__/my_solution.cpython-39.pyc CHANGED
Binary files a/__pycache__/my_solution.cpython-39.pyc and b/__pycache__/my_solution.cpython-39.pyc differ
 
my_solution.py CHANGED
@@ -61,4 +61,6 @@ def predict(entry, visualize=False) -> Tuple[np.ndarray, List[int]]:
61
  edges,
62
  entry['wf_vertices'],
63
  entry['wf_edges'])
 
 
64
  return entry['__key__'], vertices, edges
 
61
  edges,
62
  entry['wf_vertices'],
63
  entry['wf_edges'])
64
+ if vertices.shape[-1] != 3:
65
+ vertices, edges = my_empty_solution()
66
  return entry['__key__'], vertices, edges