nshtrainer 0.10.2__py3-none-any.whl → 0.10.4__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 (5) hide show
  1. nshtrainer/callbacks/wandb_watch.py +1 -1
  2. nshtrainer/model/_environment.py +3 -3
  3. {nshtrainer-0.10.2.dist-info → nshtrainer-0.10.4.dist-info}/METADATA +1 -1
  4. {nshtrainer-0.10.2.dist-info → nshtrainer-0.10.4.dist-info}/RECORD +5 -5
  5. {nshtrainer-0.10.2.dist-info → nshtrainer-0.10.4.dist-info}/WHEEL +0 -0
nshtrainer/callbacks/wandb_watch.py CHANGED
@@ -81,7 +81,7 @@ class WandbWatchCallback(Callback):
81
81
 
82
82
 
83
83
  class WandbWatchConfig(CallbackConfigBase):
84
- name: Literal["finite_checks"] = "finite_checks"
84
+ name: Literal["wandb_watch"] = "wandb_watch"
85
85
 
86
86
  enabled: bool = True
87
87
  """Enable watching the model for wandb."""
nshtrainer/model/_environment.py CHANGED
@@ -409,9 +409,9 @@ class EnvironmentPackageConfig(C.Config):
409
409
  name=package.project_name,
410
410
  version=package.version,
411
411
  path=Path(package.location) if package.location else None,
412
- summary=package.summary,
413
- author=package.author,
414
- license=package.license,
412
+ summary=getattr(package, "summary", None),
413
+ author=getattr(package, "author", None),
414
+ license=getattr(package, "license", None),
415
415
  requires=[str(req) for req in package.requires()],
416
416
  )
417
417
  except ImportError:
{nshtrainer-0.10.2.dist-info → nshtrainer-0.10.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.10.2
3
+ Version: 0.10.4
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.10.2.dist-info → nshtrainer-0.10.4.dist-info}/RECORD RENAMED
@@ -22,7 +22,7 @@ nshtrainer/callbacks/on_exception_checkpoint.py,sha256=x42BYZ2ejf2rhqPLCmT5nyWKh
22
22
  nshtrainer/callbacks/print_table.py,sha256=_FdAHhqylWGk4Z0c2FrLFeiMA4jhfA_beZRK_BHpzmE,2837
23
23
  nshtrainer/callbacks/throughput_monitor.py,sha256=4EF3b79HdHiRgBGIFDyD4O1oywb5h1tV8nml7NuuDjU,1845
24
24
  nshtrainer/callbacks/timer.py,sha256=quS79oYClDUvQxJkNWmDMe0hwRUkkREgTgqzVrnom50,4607
25
- nshtrainer/callbacks/wandb_watch.py,sha256=bicXS3nZfPGoN7Owu1XIBS-1bw7yeIJdYJTnRN0dp2E,2934
25
+ nshtrainer/callbacks/wandb_watch.py,sha256=0JKDEPSDmiKverFm00lFfufOvtvr49akFXScvdUQnqc,2930
26
26
  nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
27
27
  nshtrainer/data/balanced_batch_sampler.py,sha256=bcJBcQjh1hB1yKF_xSlT9AtEWv0BJjYc1CuH2BF-ea8,4392
28
28
  nshtrainer/data/transform.py,sha256=JeGxvytQly8hougrsdMmKG8gJ6qvFPDglJCO4Tp6STk,1795
@@ -50,7 +50,7 @@ nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=h76oTHYpMxauV_l6lviya5DW-
50
50
  nshtrainer/metrics/__init__.py,sha256=ObLIELGguIEcUpRsUkqh1ltrvZii6vglTpJGrPvoy00,50
51
51
  nshtrainer/metrics/_config.py,sha256=hWWS4IXENRyH3RmJ7z1Wx1n3Lt1sNMlGOrcU6PW15o0,1104
52
52
  nshtrainer/model/__init__.py,sha256=TbexTxiE20WHYg5q3L88Hysk4LlHeKk_isv33aSBREA,1918
53
- nshtrainer/model/_environment.py,sha256=s3JFnigbssFRJTwH33K7DcAYVhLOFCC1OZgFNXJgjuw,22317
53
+ nshtrainer/model/_environment.py,sha256=aKr4S-Hp63TAPgr5qC6I5E-MYjzZ_puRaJT75ZAgqJI,22371
54
54
  nshtrainer/model/base.py,sha256=Bmw-t70TydDbE9P0ee-lTibGoUhrCx5Qke-upa7FGVM,17512
55
55
  nshtrainer/model/config.py,sha256=OsVba02cmEYVf6V-A6ljV7VMAW5XZO6GWNRk8ktUw2o,53177
56
56
  nshtrainer/model/modules/callback.py,sha256=JF59U9-CjJsAIspEhTJbVaGN0wGctZG7UquE3IS7R8A,6408
@@ -79,6 +79,6 @@ nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
79
79
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
80
80
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
81
81
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
82
- nshtrainer-0.10.2.dist-info/METADATA,sha256=bLP7xa9qV9BqgqGZr2RayfaL2eFB9eBSTHVlMpslm5s,695
83
- nshtrainer-0.10.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
84
- nshtrainer-0.10.2.dist-info/RECORD,,
82
+ nshtrainer-0.10.4.dist-info/METADATA,sha256=xyxTCvbe5YqMiwXisCYgvEYIdZVkgw-gtb8gVB_Ptjc,695
83
+ nshtrainer-0.10.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
84
+ nshtrainer-0.10.4.dist-info/RECORD,,