molcraft 0.1.0a19__tar.gz → 0.1.0a20__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of molcraft might be problematic. Click here for more details.

Files changed (32) hide show
  1. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/PKG-INFO +1 -1
  2. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/__init__.py +1 -1
  3. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/featurizers.py +2 -1
  4. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft.egg-info/PKG-INFO +1 -1
  5. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/tests/test_featurizers.py +14 -0
  6. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/LICENSE +0 -0
  7. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/README.md +0 -0
  8. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/applications/__init__.py +0 -0
  9. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/applications/chromatography.py +0 -0
  10. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/applications/proteomics.py +0 -0
  11. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/callbacks.py +0 -0
  12. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/chem.py +0 -0
  13. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/datasets.py +0 -0
  14. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/descriptors.py +0 -0
  15. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/features.py +0 -0
  16. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/layers.py +0 -0
  17. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/losses.py +0 -0
  18. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/models.py +0 -0
  19. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/ops.py +0 -0
  20. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/records.py +0 -0
  21. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft/tensors.py +0 -0
  22. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft.egg-info/SOURCES.txt +0 -0
  23. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft.egg-info/dependency_links.txt +0 -0
  24. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft.egg-info/requires.txt +0 -0
  25. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/molcraft.egg-info/top_level.txt +0 -0
  26. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/pyproject.toml +0 -0
  27. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/setup.cfg +0 -0
  28. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/tests/test_chem.py +0 -0
  29. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/tests/test_layers.py +0 -0
  30. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/tests/test_losses.py +0 -0
  31. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/tests/test_models.py +0 -0
  32. {molcraft-0.1.0a19 → molcraft-0.1.0a20}/tests/test_tensors.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: molcraft
3
- Version: 0.1.0a19
3
+ Version: 0.1.0a20
4
4
  Summary: Graph Neural Networks for Molecular Machine Learning
5
5
  Author-email: Alexander Kensert <alexander.kensert@gmail.com>
6
6
  License: MIT License
@@ -1,4 +1,4 @@
1
- __version__ = '0.1.0a19'
1
+ __version__ = '0.1.0a20'
2
2
 
3
3
  import os
4
4
  os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
@@ -323,7 +323,9 @@ class MolGraphFeaturizer3D(MolGraphFeaturizer):
323
323
  include_hydrogens: bool = False,
324
324
  radius: int | float | None = 6.0,
325
325
  random_seed: int | None = None,
326
+ **kwargs,
326
327
  ) -> None:
328
+ kwargs.pop('bond_features', None)
327
329
  super().__init__(
328
330
  atom_features=atom_features,
329
331
  bond_features=None,
@@ -413,7 +415,6 @@ class MolGraphFeaturizer3D(MolGraphFeaturizer):
413
415
 
414
416
  def get_config(self):
415
417
  config = super().get_config()
416
- config.pop('bond_features', None)
417
418
  config['radius'] = self._radius
418
419
  config['pair_features'] = keras.saving.serialize_keras_object(
419
420
  self._pair_features
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: molcraft
3
- Version: 0.1.0a19
3
+ Version: 0.1.0a20
4
4
  Summary: Graph Neural Networks for Molecular Machine Learning
5
5
  Author-email: Alexander Kensert <alexander.kensert@gmail.com>
6
6
  License: MIT License
@@ -1,4 +1,6 @@
1
1
  import unittest
2
+ import tempfile
3
+ import shutil
2
4
 
3
5
  from molcraft import features
4
6
  from molcraft import featurizers
@@ -42,6 +44,12 @@ class TestFeaturizer(unittest.TestCase):
42
44
  include_hydrogens=False,
43
45
  )
44
46
 
47
+ tmp_dir = tempfile.mkdtemp()
48
+ tmp_file = tmp_dir + '/featurizer.json'
49
+ featurizers.save_featurizer(featurizer, tmp_file)
50
+ _ = featurizers.load_featurizer(tmp_file)
51
+ shutil.rmtree(tmp_dir)
52
+
45
53
  node_dim = 9
46
54
  edge_dim = 4
47
55
 
@@ -127,6 +135,12 @@ class TestFeaturizer(unittest.TestCase):
127
135
  radius=5.0,
128
136
  )
129
137
 
138
+ tmp_dir = tempfile.mkdtemp()
139
+ tmp_file = tmp_dir + '/featurizer.json'
140
+ featurizers.save_featurizer(featurizer, tmp_file)
141
+ _ = featurizers.load_featurizer(tmp_file)
142
+ shutil.rmtree(tmp_dir)
143
+
130
144
  node_dim = 10
131
145
  edge_dim = 22
132
146
 
File without changes
File without changes
File without changes
File without changes
File without changes