Merge pull request #13 from lucidrains/vector-gating

allow one to set their own path to cache directory with environmental…

Author: lucidrains

README.md

@@ -520,6 +520,12 @@ Or you can try deleting the cache directory, which should exist at
 $ rm -rf ~/.cache.equivariant_attention
 ```
 
+You can also designate your own directory where you want the caches to be stored, in the case that the default directory may have permission issues
+
+```bash
+CACHE_PATH=./path/to/my/cache python train.py
+```
+
 ## Testing
 
 ```bash

se3_transformer_pytorch/basis.py

@@ -7,12 +7,13 @@
 from contextlib import contextmanager
 
 from se3_transformer_pytorch.irr_repr import irr_repr, spherical_harmonics
-from se3_transformer_pytorch.utils import torch_default_dtype, cache_dir, exists, to_order
+from se3_transformer_pytorch.utils import torch_default_dtype, cache_dir, exists, default, to_order
 from se3_transformer_pytorch.spherical_harmonics import clear_spherical_harmonics_cache
 
 # constants
 
-CACHE_PATH = os.path.expanduser('~/.cache.equivariant_attention') if not exists(os.environ.get('CLEAR_CACHE')) else None
+CACHE_PATH = default(os.getenv('CACHE_PATH'), os.path.expanduser('~/.cache.equivariant_attention'))
+CACHE_PATH = CACHE_PATH if not exists(os.environ.get('CLEAR_CACHE')) else None
 
 # todo (figure ot why this was hard coded in official repo)
 

setup.py

@@ -4,7 +4,7 @@
   name = 'se3-transformer-pytorch',
   packages = find_packages(),
   include_package_data = True,
-  version = '0.8.7',
+  version = '0.8.8',
   license='MIT',
   description = 'SE3 Transformer - Pytorch',
   author = 'Phil Wang',