From 189abf4d1e51221c7e8edb05b36c0b41272e583b Mon Sep 17 00:00:00 2001
From: Amogh Joshi <joshi.amoghn@gmail.com>
Date: Thu, 4 Jan 2024 14:13:05 -0500
Subject: [PATCH] Push to version 0.6.0

---
 agml/__init__.py |  2 +-
 tests/main.py    | 10 ++++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/agml/__init__.py b/agml/__init__.py
index f8c78bb5..389984f8 100644
--- a/agml/__init__.py
+++ b/agml/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '0.5.2'
+__version__ = '0.6.0'
 __all__ = ['data', 'synthetic', 'backend', 'viz', 'io']
 
 
diff --git a/tests/main.py b/tests/main.py
index 843b97bc..2887597f 100644
--- a/tests/main.py
+++ b/tests/main.py
@@ -21,6 +21,7 @@
 _USED_DATASETS = ['bean_disease_uganda', 'apple_flower_segmentation', 'apple_detection_usa']
 _PREEXISTING_DATASETS = []
 
+
 def _find_preexisting_datasets():
     global _USED_DATASETS, _PREEXISTING_DATASETS, BASE_PATH
     for dataset in _USED_DATASETS:
@@ -28,26 +29,27 @@ def _find_preexisting_datasets():
                 and os.path.isdir(os.path.join(BASE_PATH, dataset)):
             _PREEXISTING_DATASETS.append(dataset)
 
+
 def _remove_new_datasets():
     global _USED_DATASETS, _PREEXISTING_DATASETS, BASE_PATH
     for dataset in _USED_DATASETS:
         if dataset not in _PREEXISTING_DATASETS:
             shutil.rmtree(dataset)
 
+
 def pre_test_configure():
     _find_preexisting_datasets()
 
+
 def post_test_cleanup():
     _remove_new_datasets()
 
+
 def execute_tests():
     pre_test_configure()
     pytest.main()
     post_test_cleanup()
 
+
 if __name__ == '__main__':
     execute_tests()
-
-
-
-