Include model parameter placeholders in site-packages (#500)

* skip downloading into site-packages

* add placeholder files

* add placeholder files

* dont check filesize in local_dir

* fix shutil erro

* use distutils instead of shutils

.gitignore

@@ -30,9 +30,6 @@ dist
 *.swo
 /download
 /download.tar.xz
-/torchani/resources/*.info
-/torchani/resources/ani-1ccx_8x
-/torchani/resources/ani-2x_8x
 *.qdrep
 *.qdstrm
 *.zip

torchani/models.py

@@ -31,10 +31,9 @@ directly calculate energies or get an ASE calculator. For example:
 import os
 import io
 import requests
-import glob
 import zipfile
-import shutil
 import torch
+from distutils import dir_util
 from torch import Tensor
 from typing import Tuple, Optional
 from . import neurochem
@@ -89,7 +88,7 @@ class BuiltinModel(torch.nn.Module):
         extracted_name = '{}-{}'.format(repo_name, tag_name)
         url = "https://github.com/aiqm/{}/archive/{}.zip".format(repo_name, tag_name)
 
-        if not os.path.isfile(get_resource(resource_path, info_file_path)):
+        if os.stat(get_resource(resource_path, info_file_path)).st_size == 0:
             if not os.path.isfile(get_resource(local_dir, info_file_path)):
                 print('Downloading ANI model parameters ...')
                 resource_res = requests.get(url)
@@ -99,17 +98,14 @@ class BuiltinModel(torch.nn.Module):
                 except PermissionError:
                     resource_zip.extractall(local_dir)
                     resource_path = local_dir
+
+                source = os.path.join(resource_path, extracted_name, "resources")
+                dir_util.copy_tree(source, resource_path)
+                dir_util.remove_tree(os.path.join(resource_path, extracted_name))
+
             else:
                 resource_path = local_dir
 
-            files = glob.glob(os.path.join(resource_path, extracted_name, "resources", "*"))
-            for f in files:
-                try:
-                    shutil.move(f, resource_path)
-                except shutil.Error:
-                    pass
-            shutil.rmtree(os.path.join(resource_path, extracted_name))
-
         info_file = get_resource(resource_path, info_file_path)
 
         with open(info_file) as f: