FreyrS / dMaSIF

Other
193 stars 45 forks source link

Tensor triangles must be of type int64. #11

Closed YannisBilly closed 1 year ago

YannisBilly commented 3 years ago

Written a test script for a cube with a triangular mesh form. The triangles were passed into the save_vtk function as torch.int16 and torch.int32 and produced the same result:

Traceback (most recent call last):
  File "custom_geometry_value_placement.py", line 31, in <module>
    values = values)
  File "/home/tsoni/workspace/athena/drug_discovery/dMaSIF/geometry_processing.py", line 43, in save_vtk
    structure = PolyData(points=numpy(xyz), polygons=numpy(triangles))
  File "/home/tsoni/workspace/athena/venvs/drug_discovery/lib/python3.7/site-packages/pyvtk/PolyData.py", line 47, in __init__
    raise ValueError('%s must be (seq of seq|seq) integers less than %s'%(k,sz))
ValueError: polygons must be (seq of seq|seq) integers less than 8

Added an assertion error for the case that the user inserts a tensor of type 64 for the triangles.