dsa-helpers 1.1.19.dev1__tar.gz → 1.1.19.dev2__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 (55) hide show
  1. {dsa_helpers-1.1.19.dev1/dsa_helpers.egg-info → dsa_helpers-1.1.19.dev2}/PKG-INFO +1 -1
  2. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/__init__.py +1 -1
  3. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/train.py +8 -0
  4. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2/dsa_helpers.egg-info}/PKG-INFO +1 -1
  5. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/pyproject.toml +1 -1
  6. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/.gitignore +0 -0
  7. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/.readthedocs.yaml +0 -0
  8. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/LICENSE +0 -0
  9. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/README.md +0 -0
  10. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/app.py +0 -0
  11. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dev.py +0 -0
  12. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/docker/Dockerfile +0 -0
  13. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/docker/README.md +0 -0
  14. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/docs/requirements.txt +0 -0
  15. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/docs/source/conf.py +0 -0
  16. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/docs/source/index.rst +0 -0
  17. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/cli_utils.py +0 -0
  18. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/dash/__init__.py +0 -0
  19. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/dash/dash_paperdragon_utils.py +0 -0
  20. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/dash/header.py +0 -0
  21. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/dash/header_callbacks.py +0 -0
  22. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/girder_utils.py +0 -0
  23. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/gpd_utils.py +0 -0
  24. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/image_utils.py +0 -0
  25. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/imread.py +0 -0
  26. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/imwrite.py +0 -0
  27. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/__init__.py +0 -0
  28. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/callbacks.py +0 -0
  29. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/datasets/SegFormerSegmentationDataset.py +0 -0
  30. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/datasets/__init__.py +0 -0
  31. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/datasets/utils.py +0 -0
  32. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/evaluate.py +0 -0
  33. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/metrics.py +0 -0
  34. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/object_detection/__init__.py +0 -0
  35. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/object_detection/get_pearce_roi_images.py +0 -0
  36. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/object_detection/tile_image.py +0 -0
  37. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/object_detection/utils.py +0 -0
  38. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/__init__.py +0 -0
  39. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/datasets.py +0 -0
  40. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/evaluate.py +0 -0
  41. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/inference.py +0 -0
  42. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/transforms.py +0 -0
  43. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/segformer_semantic_segmentation/utils.py +0 -0
  44. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/transforms/__init__.py +0 -0
  45. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/ml/transforms/segformer_transforms.py +0 -0
  46. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/mongo_utils.py +0 -0
  47. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/tile_utils.py +0 -0
  48. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/tiling.py +0 -0
  49. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers/utils.py +0 -0
  50. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers.egg-info/SOURCES.txt +0 -0
  51. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers.egg-info/dependency_links.txt +0 -0
  52. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers.egg-info/requires.txt +0 -0
  53. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/dsa_helpers.egg-info/top_level.txt +0 -0
  54. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/requirements.txt +0 -0
  55. {dsa_helpers-1.1.19.dev1 → dsa_helpers-1.1.19.dev2}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dsa_helpers
3
- Version: 1.1.19.dev1
3
+ Version: 1.1.19.dev2
4
4
  Summary: Utility functions for working with the DSA girder client.
5
5
  Author-email: "Juan C. Vizcarra" <jvizcar@emory.edu>
6
6
  Maintainer-email: "Juan C. Vizcarra" <jvizcar@emory.edu>
@@ -3,7 +3,7 @@ from .imread import imread
3
3
  from .imwrite import imwrite
4
4
 
5
5
  # Version of the dsa-helpers package
6
- __version__ = "1.1.19dev1"
6
+ __version__ = "1.1.19dev2"
7
7
 
8
8
  # To avoid slow downs, do not allow from dsa_helpers import * to import anything.
9
9
  __all__ = []
@@ -6,6 +6,7 @@ from transformers import (
6
6
  )
7
7
  import numpy as np
8
8
  from pathlib import Path
9
+ import tempfile, shutil
9
10
 
10
11
  import torch
11
12
  from torch import nn
@@ -144,6 +145,10 @@ def train(
144
145
  ignore_mismatched_sizes=True,
145
146
  ).to(device)
146
147
 
148
+ if rank != 0:
149
+ # Create temp directory for the model.
150
+ save_dir = tempfile.mkdtemp()
151
+
147
152
  # Training arguments.
148
153
  training_args = TrainingArguments(
149
154
  str(save_dir),
@@ -258,6 +263,9 @@ def train(
258
263
  if rank != 0:
259
264
  # Train the model.
260
265
  _ = trainer.train()
266
+
267
+ # Delete the temp directory.
268
+ shutil.rmtree(save_dir)
261
269
  return None
262
270
  else:
263
271
  print("Starting training...")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dsa_helpers
3
- Version: 1.1.19.dev1
3
+ Version: 1.1.19.dev2
4
4
  Summary: Utility functions for working with the DSA girder client.
5
5
  Author-email: "Juan C. Vizcarra" <jvizcar@emory.edu>
6
6
  Maintainer-email: "Juan C. Vizcarra" <jvizcar@emory.edu>
@@ -41,7 +41,7 @@ classifiers = [
41
41
  "License :: OSI Approved :: MIT License",
42
42
  ]
43
43
  keywords = ["dsa", "girder", "girder_client"]
44
- version = "v1.1.19dev1"
44
+ version = "v1.1.19dev2"
45
45
 
46
46
  [project.optional-dependencies]
47
47
  opencv = ["opencv-python"]