resolve flake8 issues

Author: MattShirley

code_generator_TopCPToolkit/tests/test_custom_image_tag.py

@@ -88,4 +88,4 @@ def test_generated_files_created_with_image_tag(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

servicex_app/servicex_app/resources/transformation/submit.py

@@ -58,16 +58,15 @@ def validate_custom_image_tag(image_tag: str, config: dict, codegen_name: str) -
         return None, None
 
     # Get validation configuration (with defaults)
-    allowed_repos = config.get('TOPCP_ALLOWED_REPOSITORIES', [
-        'sslhep/servicex_science_image_topcp',
-        'registry.gitlab.com/topcp-project/toolkit'
-    ])
-    tag_pattern = config.get('TOPCP_IMAGE_TAG_PATTERN', r'^v?\d+\.\d+\.\d+[-_]v?\d+\.\d+$')
-    default_base_image = config.get('TOPCP_DEFAULT_BASE_IMAGE', 'sslhep/servicex_science_image_topcp')
+    tag_pattern = config.get('TOPCP_IMAGE_TAG_PATTERN',
+                             r'^v?\d+\.\d+\.\d+[-_]v?\d+\.\d+$')
+    default_base_image = config.get('TOPCP_DEFAULT_BASE_IMAGE',
+                                    'sslhep/servicex_science_image_topcp')
 
     # Validate tag format
     if not re.match(tag_pattern, image_tag):
-        raise BadRequest(f"Invalid TopCP image tag format: {image_tag}. Expected format: v2.20.0_v0.2")
+        raise BadRequest(f"Invalid TopCP image tag format: {image_tag}. "
+                         f"Expected format: v2.20.0_v0.2")
 
     # For now, use the default base image with the custom tag
     # In the future, this could be extended to allow custom repositories
@@ -97,6 +96,7 @@ def _extract_custom_image_tag(self, selection: str, codegen_name: str) -> Option
         except (json.JSONDecodeError, AttributeError):
             # Invalid JSON or non-dict selection
             return None
+
     @classmethod
     def make_api(
         cls,
@@ -270,7 +270,8 @@ def post(self):
             )
 
             # Extract custom image tag from TopCP queries before code generation
-            custom_image_tag = self._extract_custom_image_tag(request_rec.selection, user_codegen_name)
+            custom_image_tag = self._extract_custom_image_tag(
+                request_rec.selection, user_codegen_name)
 
             # The first thing to do is make sure the requested selection is correct,
             # and can generate the requested code

servicex_app/servicex_app_test/test_custom_image_validation.py

@@ -4,7 +4,9 @@
 import unittest
 from werkzeug.exceptions import BadRequest
 
-from servicex_app.resources.transformation.submit import validate_custom_image_tag, SubmitTransformationRequest
+from servicex_app.resources.transformation.submit import (
+    validate_custom_image_tag, SubmitTransformationRequest
+)
 
 
 class TestCustomImageValidation(unittest.TestCase):
@@ -138,4 +140,4 @@ def test_extract_image_tag_none_value(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()