kernel-elastic-autoencoder 3.4.0__tar.gz → 3.4.1__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.
Files changed (11) hide show
  1. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/PKG-INFO +1 -1
  2. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/pyproject.toml +1 -1
  3. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/pipeline.py +2 -6
  4. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/README.md +0 -0
  5. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/__init__.py +0 -0
  6. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/config.py +0 -0
  7. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/layers.py +0 -0
  8. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/losses.py +0 -0
  9. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/model.py +0 -0
  10. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/tokenizer.py +0 -0
  11. {kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/training.py +0 -0
{kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: kernel_elastic_autoencoder
3
- Version: 3.4.0
3
+ Version: 3.4.1
4
4
  Summary: Implementation of Kernel-Elastic Autoencoder for Molecular Design (https://doi.org/10.1093/pnasnexus/pgae168)
5
5
  License: MIT
6
6
  Author: Felix Rotter-McCartney
{kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "kernel_elastic_autoencoder"
3
- version = "3.4.0"
3
+ version = "3.4.1"
4
4
  description = "Implementation of Kernel-Elastic Autoencoder for Molecular Design (https://doi.org/10.1093/pnasnexus/pgae168)"
5
5
  authors = [
6
6
  { name = "Felix Rotter-McCartney", email = "felix.rotter@mail.utoronto.ca" }
{kernel_elastic_autoencoder-3.4.0 → kernel_elastic_autoencoder-3.4.1}/src/kernel_elastic_autoencoder/pipeline.py RENAMED
@@ -265,9 +265,7 @@ class Pipeline:
265
265
  top_prob_inds = (
266
266
  (
267
267
  top_probs.sum(dim=-1)
268
- / torch.sqrt(grouped_ids != self.tokenizer.pad_token_id)
269
- .to(torch.long)
270
- .sum(dim=-1)
268
+ / torch.sqrt((grouped_ids != self.tokenizer.pad_token_id).to(torch.long).sum(dim=-1))
271
269
  )
272
270
  .topk(k=beam_size, dim=1)
273
271
  .indices.squeeze(-1)
@@ -297,9 +295,7 @@ class Pipeline:
297
295
  top_prob_inds = (
298
296
  (
299
297
  top_probs.sum(dim=-1)
300
- / torch.sqrt(grouped_ids != self.tokenizer.pad_token_id)
301
- .to(torch.long)
302
- .sum(dim=-1)
298
+ / torch.sqrt((grouped_ids != self.tokenizer.pad_token_id).to(torch.long).sum(dim=-1))
303
299
  )
304
300
  .topk(k=1, dim=1)
305
301
  .indices.squeeze(-1)