-
Gao, Xiang authored
* Specify dtype for species For some reason, sometimes species are constructed as int tensors instead of long tensors, this is wrong and should be fixed * Update __init__.py
42442af6
* Specify dtype for species For some reason, sometimes species are constructed as int tensors instead of long tensors, this is wrong and should be fixed * Update __init__.py