• Ignacio Pickering's avatar
    Tests cleanup (#495) · 28ecce79
    Ignacio Pickering authored
    * Remove test gradient from aev tests
    
    * Add a dedicated test gradient for one model, and make it feasible
    
    * add test_grad to unittests
    
    * Add some comments and fix formatting
    
    * flake8
    
    * flake8
    
    * remove unused var
    
    * delete transform function, which was legacy from old pytorch probably
    
    * Delete random_skip function, which was unused
    
    * Allow testNeuroChem to use cuda if available, which makes it much faster
    
    * small modification for consistency
    
    * remove unused import
    
    * simplify device call
    
    * remove unused variable
    
    * add main to test_grad
    28ecce79
test_aev_benzene_md.py 1.21 KB