synapse-sdk 1.0.0b5__py3-none-any.whl → 1.0.0b6__py3-none-any.whl

synapse_sdk/plugins/categories/export/actions/export.py

@@ -1,6 +1,7 @@
 import json
 from abc import ABC, abstractmethod
 from datetime import datetime
+from itertools import tee
 from typing import Annotated, Any, Literal
 
 from pydantic import AfterValidator, BaseModel, field_validator
@@ -324,16 +325,18 @@ class ExportAction(Action):
         handler = TargetHandlerFactory.get_handler(target)
 
         self.params['results'], self.params['count'] = self.get_filtered_results(filters, handler)
-        export_items = handler.get_export_item(self.params['results'])
 
         # For the 'ground_truth' target, retrieve project information from the first result and add configuration
         if target == 'ground_truth':
             try:
-                first_result = next(iter(self.params['results']))
+                # Split generator into two using tee()
+                peek_iter, main_iter = tee(self.params['results'])
+                first_result = next(peek_iter)  # Peek first value only
                 project_pk = first_result['project']
                 project_info = self.client.get_project(project_pk)
                 self.params['configuration'] = project_info.get('configuration', {})
-            except StopIteration:
+                self.params['results'] = main_iter  # Keep original generator intact
+            except (StopIteration, KeyError):
                 self.params['configuration'] = {}
         # For the 'assignment' and 'task' targets, retrieve the project from the filter as before
         elif target in ['assignment', 'task'] and 'project' in self.params['filter']:
@@ -341,6 +344,7 @@ class ExportAction(Action):
             project_info = self.client.get_project(project_pk)
             self.params['configuration'] = project_info.get('configuration', {})
 
+        export_items = handler.get_export_item(self.params['results'])
         storage = self.client.get_storage(self.params['storage'])
         pathlib_cwd = get_pathlib(storage, self.params['path'])
         return self.entrypoint(self.run, export_items, pathlib_cwd, **self.params)

{synapse_sdk-1.0.0b5.dist-info → synapse_sdk-1.0.0b6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0b5
+Version: 1.0.0b6
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0b5.dist-info → synapse_sdk-1.0.0b6.dist-info}/RECORD

@@ -113,7 +113,7 @@ synapse_sdk/plugins/categories/data_validation/templates/plugin/validation.py,sh
 synapse_sdk/plugins/categories/export/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/export/enums.py,sha256=gtyngvQ1DKkos9iKGcbecwTVQQ6sDwbrBPSGPNb5Am0,127
 synapse_sdk/plugins/categories/export/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/plugins/categories/export/actions/export.py,sha256=7OCU3kwkYqYnelG6vfZ4ClyyMj_FHT-WQoUmgcfH080,12012
+synapse_sdk/plugins/categories/export/actions/export.py,sha256=XxhFIebt9tGTNcj0N6guVhAPP4d2YZ5Lvsx22-va72Y,12263
 synapse_sdk/plugins/categories/export/templates/config.yaml,sha256=N7YmnFROb3s3M35SA9nmabyzoSb5O2t2TRPicwFNN2o,56
 synapse_sdk/plugins/categories/export/templates/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/export/templates/plugin/export.py,sha256=GDb6Ucodsr5aBPMU4alr68-DyFoLR5TyhC_MCaJrkF0,6411
@@ -211,9 +211,9 @@ synapse_sdk/utils/storage/providers/gcp.py,sha256=i2BQCu1Kej1If9SuNr2_lEyTcr5M_n
 synapse_sdk/utils/storage/providers/http.py,sha256=2DhIulND47JOnS5ZY7MZUex7Su3peAPksGo1Wwg07L4,5828
 synapse_sdk/utils/storage/providers/s3.py,sha256=ZmqekAvIgcQBdRU-QVJYv1Rlp6VHfXwtbtjTSphua94,2573
 synapse_sdk/utils/storage/providers/sftp.py,sha256=_8s9hf0JXIO21gvm-JVS00FbLsbtvly4c-ETLRax68A,1426
-synapse_sdk-1.0.0b5.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0b5.dist-info/METADATA,sha256=dE3YqjlFZ96vvRoQ59Htr65rSrvmibKDAvsGwOIB6U4,3718
-synapse_sdk-1.0.0b5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-synapse_sdk-1.0.0b5.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0b5.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0b5.dist-info/RECORD,,
+synapse_sdk-1.0.0b6.dist-info/licenses/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0b6.dist-info/METADATA,sha256=mU8-WwXY45NYGlXdk_mS1P9P8Ip21Q_9RqLZ9joe9hc,3718
+synapse_sdk-1.0.0b6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synapse_sdk-1.0.0b6.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0b6.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0b6.dist-info/RECORD,,