restore sidecar capabilities

ssl-hep · May 30, 2023 · e401e12 · e401e12
1 parent 6f26d35
commit e401e12
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/tests/test_post_operation.py b/tests/test_post_operation.py
@@ -66,7 +66,8 @@ def test_post_good_query_with_params(self, mocker):
             )
 
             config = {
-                'TARGET_BACKEND': 'uproot'
+                'TARGET_BACKEND': 'uproot',
+                'TRANSFORMER_SCIENCE_IMAGE': "foo/bar:latest"
             }
             app = create_app(config, provided_translator=mock_ast_translator)
             client = app.test_client()
@@ -85,7 +86,7 @@ def test_post_good_query_with_params(self, mocker):
         decoder_parts = decoder.MultipartDecoder(response.data, content_type)
 
         transformer_image = str(decoder_parts.parts[0].content, 'utf-8')
-        zip_file = decoder_parts.parts[1].content
+        zip_file = decoder_parts.parts[3].content
 
         print("Transformer Image: ", transformer_image)
         print("Zip File: ", zip_file)
@@ -130,7 +131,7 @@ def test_post_good_query_without_params(self, mocker):
             decoder_parts = decoder.MultipartDecoder(response.data, content_type)
 
             transformer_image = str(decoder_parts.parts[0].content, 'utf-8')
-            zip_file = decoder_parts.parts[1].content
+            zip_file = decoder_parts.parts[3].content
 
             print("Transformer Image: ", transformer_image)
             print("Zip File: ", zip_file)

diff --git a/transformer_capabilities.json b/transformer_capabilities.json
@@ -0,0 +1,9 @@
+{
+  "name": "FuncADL based uproot transformer",
+  "description": "Extracts data from flat ntuple style root files.",
+  "limitations": "Would be good to note what isn't implemented",
+  "file-formats": ["parquet"],
+  "stats-parser": "UprootStats",
+  "language": "python",
+  "command": "/generated/transform_single_file.py"
+}