From 00d65a62c9ea21ed50726766ec36cd665e7a7e7a Mon Sep 17 00:00:00 2001 From: James McKinney <26463+jpmckinney@users.noreply.github.com> Date: Fri, 27 Sep 2024 16:56:16 -0400 Subject: [PATCH] chore(style): Use set for set inclusion condition --- ocdsextensionregistry/extension_version.py | 2 +- ocdsextensionregistry/profile_builder.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ocdsextensionregistry/extension_version.py b/ocdsextensionregistry/extension_version.py index 924cc1e..7459221 100644 --- a/ocdsextensionregistry/extension_version.py +++ b/ocdsextensionregistry/extension_version.py @@ -129,7 +129,7 @@ def files(self): filename = name[start:] if filename[-1] != '/' and not filename.startswith('.'): content = zipfile.read(name) - if os.path.splitext(name)[1] in ('.csv', '.json', '.md'): + if os.path.splitext(name)[1] in {'.csv', '.json', '.md'}: content = content.decode('utf-8') files[filename] = content diff --git a/ocdsextensionregistry/profile_builder.py b/ocdsextensionregistry/profile_builder.py index 78763c5..a03ee7e 100644 --- a/ocdsextensionregistry/profile_builder.py +++ b/ocdsextensionregistry/profile_builder.py @@ -365,7 +365,7 @@ def _add_extension_field(schema, extension_name, field_name, pointer=None): for item in schema: _add_extension_field(item, extension_name, field_name=field_name, pointer=pointer) elif isinstance(schema, dict): - if len(pointer) > 1 and pointer[-2] in ('definitions', 'properties') and 'title' in schema: + if len(pointer) > 1 and pointer[-2] in {'definitions', 'properties'} and 'title' in schema: schema[field_name] = extension_name for key, value in schema.items(): _add_extension_field(value, extension_name, field_name=field_name, pointer=(*pointer, key))