heavyball 1.1.2__py3-none-any.whl → 1.1.3__py3-none-any.whl

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 (7) hide show
  1. heavyball/chainable.py +3 -4
  2. {heavyball-1.1.2.dist-info → heavyball-1.1.3.dist-info}/METADATA +1 -1
  3. heavyball-1.1.3.dist-info/RECORD +8 -0
  4. heavyball-1.1.2.dist-info/RECORD +0 -8
  5. {heavyball-1.1.2.dist-info → heavyball-1.1.3.dist-info}/LICENSE +0 -0
  6. {heavyball-1.1.2.dist-info → heavyball-1.1.3.dist-info}/WHEEL +0 -0
  7. {heavyball-1.1.2.dist-info → heavyball-1.1.3.dist-info}/top_level.txt +0 -0
heavyball/chainable.py CHANGED
@@ -1,6 +1,6 @@
1
1
  import functools
2
2
  import random
3
- from typing import Optional, Union
3
+ from typing import Optional, Union, Literal
4
4
 
5
5
  import torch
6
6
 
@@ -51,8 +51,7 @@ class FunctionTransform:
51
51
 
52
52
 
53
53
  def _zero_guard(state, key, ref, dtype):
54
- return _guard_in_state(state, key,
55
- lambda: torch.zeros_like(ref, dtype=torch.float32, memory_format=torch.preserve_format))
54
+ return _guard_in_state(state, key, lambda: torch.zeros_like(ref, dtype=dtype, memory_format=torch.preserve_format))
56
55
 
57
56
 
58
57
  def _storage_dtype(group):
@@ -442,7 +441,7 @@ class ChainOpt(utils.StatefulOptimizer):
442
441
 
443
442
 
444
443
  use_default = object()
445
- str_or_fn = Union[str, callable, None, use_default]
444
+ str_or_fn = Union[str, callable, None, Literal[use_default]]
446
445
 
447
446
 
448
447
  def _get_clip_fn(name: str_or_fn, default_val: str_or_fn):
{heavyball-1.1.2.dist-info → heavyball-1.1.3.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: heavyball
3
- Version: 1.1.2
3
+ Version: 1.1.3
4
4
  Summary: Efficient optimizers
5
5
  Home-page: https://github.com/clashluke/heavyball
6
6
  Author: Lucas Nestler
heavyball-1.1.3.dist-info/RECORD ADDED
@@ -0,0 +1,8 @@
1
+ heavyball/__init__.py,sha256=miRgcXlzLWTNzojeRF5hEcg-x_GqfMHjRzOaiR_zO3U,10981
2
+ heavyball/chainable.py,sha256=IrTlhiHvBlc1SaUUQDb1ulVfj0nrPJcqoP52AXWO3cI,20362
3
+ heavyball/utils.py,sha256=0j5wRDYeI9Elz9m8tcP7CZNhj_9OIWEF_uQpb0LTrYM,47814
4
+ heavyball-1.1.3.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
5
+ heavyball-1.1.3.dist-info/METADATA,sha256=ILjX-OviZL9vO7bgw7ldGx-8XXIa1YfWRYCW94EtAOI,12022
6
+ heavyball-1.1.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
7
+ heavyball-1.1.3.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
8
+ heavyball-1.1.3.dist-info/RECORD,,
heavyball-1.1.2.dist-info/RECORD DELETED
@@ -1,8 +0,0 @@
1
- heavyball/__init__.py,sha256=miRgcXlzLWTNzojeRF5hEcg-x_GqfMHjRzOaiR_zO3U,10981
2
- heavyball/chainable.py,sha256=Zp7q6RHYU4RgdZ_ezgc8NWPwsNfyFjRvhEK-IEqr4b4,20379
3
- heavyball/utils.py,sha256=0j5wRDYeI9Elz9m8tcP7CZNhj_9OIWEF_uQpb0LTrYM,47814
4
- heavyball-1.1.2.dist-info/LICENSE,sha256=CGdGJim64YifGmUVPaeyRsxkvyExtClswhRNIp8FY_U,1322
5
- heavyball-1.1.2.dist-info/METADATA,sha256=bhXVJpcuwNZaOKFydknhtqqYx0ZZsQp2wkEdUAoDfN4,12022
6
- heavyball-1.1.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
7
- heavyball-1.1.2.dist-info/top_level.txt,sha256=SzCxSVg_qCUPA4kZObW3Zyo4v-d_mMOD-p7a-WXTl2E,10
8
- heavyball-1.1.2.dist-info/RECORD,,