add noqa comments after running TRY

assets/scripts/build_gallery.py

@@ -108,7 +108,7 @@ def execute_shell_command(command: str) -> int:
         exception_message += (
             f'{type(cpe).__name__}: "{cpe!s}".  Traceback: "{exception_traceback}".'
         )
-        logger.error(exception_message)
+        logger.error(exception_message)  # noqa: TRY400
 
     return status_code
 
@@ -296,7 +296,7 @@ def get_expectation_instances(expectations_info):  # noqa: C901 - too complex
             try:
                 importlib.import_module(*import_module_args)
             except (ModuleNotFoundError, ImportError, Exception):
-                logger.error(f"Failed to load expectation_name: {expectation_name}")
+                logger.error(f"Failed to load expectation_name: {expectation_name}")  # noqa: TRY400
                 print(traceback.format_exc())
                 expectation_tracebacks.write(
                     f"\n\n----------------\n{expectation_name} ({expectations_info[expectation_name]['package']})\n"
@@ -311,7 +311,7 @@ def get_expectation_instances(expectations_info):  # noqa: C901 - too complex
             expectation = create_expectation_instance(expectation_class)
             expectation_instances[expectation_name] = expectation
         except ExpectationNotFoundError:
-            logger.error(
+            logger.error(  # noqa: TRY400
                 f"Failed to get Expectation implementation from registry: {expectation_name}"
             )
             print(traceback.format_exc())
@@ -348,7 +348,7 @@ def create_expectation_instance(expectation_class: type[Expectation]) -> Expecta
         ).resolve()
         found = next(_TEST_DEFS_DIR.rglob(f"**/{expectation_class.expectation_type}.json"), None)
         if not found:
-            raise Exception(
+            raise Exception(  # noqa: TRY002, TRY003
                 f"No JSON test case found for Expectation {expectation_class.expectation_type}, cannot build gallery example."
             )
         with open(found) as fp:
@@ -362,9 +362,9 @@ def get_success_test_case(tests: list[dict]) -> dict:
     try:
         return next(test["in"] for test in tests if test["out"]["success"] is True)
     except StopIteration:
-        raise ValueError("Error: Expectation test case has no valid success case.")
+        raise ValueError("Error: Expectation test case has no valid success case.")  # noqa: TRY003
     except IndexError:
-        raise ValueError("Error: Expectation test case is malformed.")
+        raise ValueError("Error: Expectation test case is malformed.")  # noqa: TRY003
 
 
 def combine_backend_results(  # noqa: C901 - too complex
@@ -399,7 +399,7 @@ def combine_backend_results(  # noqa: C901 - too complex
             try:
                 diagnostic_object = diagnostic_objects[expectation_name]
             except KeyError:
-                logger.error(f"No diagnostic obj for {expectation_name}")
+                logger.error(f"No diagnostic obj for {expectation_name}")  # noqa: TRY400
                 bad_key_names.append(expectation_name)
                 continue
             expectation_instance = expectation_instances[expectation_name]
@@ -411,7 +411,7 @@ def combine_backend_results(  # noqa: C901 - too complex
                     ]
                 ]
             except KeyError:
-                logger.error(f"No backend_test_result_counts for {expectation_name}")
+                logger.error(f"No backend_test_result_counts for {expectation_name}")  # noqa: TRY400
                 bad_key_names.append(expectation_name)
                 continue
             maturity_checklist_object = expectation_instance._get_maturity_checklist(
@@ -578,7 +578,7 @@ def build_gallery(  # noqa: C901 - 17
                 expectation_docstrings.write("\n" + "." * 80 + "\n\n")
                 expectation_docstrings.write(f"{diagnostics['description']['docstring']}\n")
         except Exception:
-            logger.error(f"Failed to run diagnostics for: {expectation_name}")
+            logger.error(f"Failed to run diagnostics for: {expectation_name}")  # noqa: TRY400
             print(traceback.format_exc())
             expectation_tracebacks.write(
                 f"\n\n----------------\n{expectation_name} ({expectations_info[expectation_name]['package']})\n"
@@ -610,7 +610,7 @@ def build_gallery(  # noqa: C901 - 17
                     }
 
             except TypeError:
-                logger.error(f"Failed to create JSON for: {expectation_name}")
+                logger.error(f"Failed to create JSON for: {expectation_name}")  # noqa: TRY400
                 print(traceback.format_exc())
                 expectation_tracebacks.write(
                     f"\n\n----------------\n[JSON write fail] {expectation_name} ({expectations_info[expectation_name]['package']})\n"

assets/scripts/build_package_gallery.py

@@ -68,7 +68,7 @@ def gather_all_package_manifests(package_paths: List[str]) -> List[dict]:
                 f"Successfully serialized {package.package_name} to dict and appended to manifest list"  # noqa: E501
             )
         except Exception as e:
-            logger.error(f"Something went wrong when syncing {path} and serializing to dict: {e}")
+            logger.error(f"Something went wrong when syncing {path} and serializing to dict: {e}")  # noqa: TRY400
         finally:
             # Always ensure we revert back to the project root
             os.chdir(root)

ci/checks/check_integration_test_gets_run.py

@@ -70,7 +70,7 @@
 def check_dependencies(*deps: str) -> None:
     for dep in deps:
         if not shutil.which(dep):
-            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")
+            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")  # noqa: TRY002, TRY003
 
 
 def get_test_files(target_dir: pathlib.Path) -> Set[str]:
@@ -98,7 +98,7 @@ def get_test_files(target_dir: pathlib.Path) -> Set[str]:
         )
         return {s.strip("/") for s in res_test_files.stdout.splitlines()}
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(
+        raise RuntimeError(  # noqa: TRY003
             f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}"
         ) from e
 
@@ -154,7 +154,7 @@ def get_test_files_in_test_suite(target_dir: pathlib.Path) -> Set[str]:
         )
         return {s.strip() for s in res_test_files_with_fixture_definitions.stdout.splitlines()}
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(
+        raise RuntimeError(  # noqa: TRY003
             f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}"
         ) from e
 

ci/checks/check_name_tag_snippets_referenced.py

@@ -352,7 +352,7 @@
 def check_dependencies(*deps: str) -> None:
     for dep in deps:
         if not shutil.which(dep):
-            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")
+            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")  # noqa: TRY002, TRY003
 
 
 def get_snippet_definitions(target_dir: pathlib.Path) -> List[str]:
@@ -380,7 +380,7 @@ def get_snippet_definitions(target_dir: pathlib.Path) -> List[str]:
         )
         return res_snippet_names.stdout.splitlines()
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(
+        raise RuntimeError(  # noqa: TRY003
             f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}"
         ) from e
 
@@ -412,7 +412,7 @@ def get_snippets_used(target_dir: pathlib.Path) -> List[str]:
         )
         return res_snippet_used_names.stdout.splitlines()
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(
+        raise RuntimeError(  # noqa: TRY003
             f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}"
         ) from e
 

ci/checks/check_no_line_number_snippets.py

@@ -19,7 +19,7 @@
 def check_dependencies(*deps: str) -> None:
     for dep in deps:
         if not shutil.which(dep):
-            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")
+            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")  # noqa: TRY002, TRY003
 
 
 def run_grep(target_dir: pathlib.Path) -> List[str]:
@@ -39,7 +39,7 @@ def run_grep(target_dir: pathlib.Path) -> List[str]:
             capture_output=True,
         )
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(
+        raise RuntimeError(  # noqa: TRY003
             f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}"
         ) from e
     return res.stdout.splitlines()

ci/checks/check_only_name_tag_snippets.py

@@ -51,7 +51,7 @@
 def check_dependencies(*deps: str) -> None:
     for dep in deps:
         if not shutil.which(dep):
-            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")
+            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")  # noqa: TRY002, TRY003
 
 
 def run_grep(target_dir: pathlib.Path) -> List[str]:
@@ -87,7 +87,7 @@ def run_grep(target_dir: pathlib.Path) -> List[str]:
             set(res_positive.stdout.splitlines()).difference(set(res_negative.stdout.splitlines()))
         )
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(
+        raise RuntimeError(  # noqa: TRY003
             f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}"
         ) from e
     return res

ci/checks/validate_docs_snippets.py

@@ -16,7 +16,7 @@
 def check_dependencies(*deps: str) -> None:
     for dep in deps:
         if not shutil.which(dep):
-            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")
+            raise Exception(f"Must have `{dep}` installed in PATH to run {__file__}")  # noqa: TRY002, TRY003
 
 
 def run_docusaurus_build(target_dir: str) -> None:
@@ -45,7 +45,7 @@ def run_grep(target_dir: str) -> List[str]:
             text=True,
         )
     except subprocess.CalledProcessError as e:
-        raise RuntimeError(f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}")
+        raise RuntimeError(f"Command {e.cmd} returned with error (code {e.returncode}): {e.output}")  # noqa: TRY003
     return res.stdout.splitlines()
 
 

contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/expectations/expect_column_values_confidence_for_data_label_to_be_greater_than_or_equal_to_threshold.py

@@ -55,7 +55,7 @@ def _pandas(
         if data_label.upper() in labeler.label_mapping.keys():
             data_label_ind = labeler.label_mapping[data_label.upper()]
         else:
-            raise ValueError(
+            raise ValueError(  # noqa: TRY003
                 """
                 The only values acceptable for the data label parameter are as follows:
                 ['PAD', 'UNKNOWN', 'ADDRESS', 'BAN', 'CREDIT_CARD', 'DATE', 'TIME', 'DATETIME',\

contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/expectations/expect_column_values_confidence_for_data_label_to_be_less_than_or_equal_to_threshold.py

@@ -55,7 +55,7 @@ def _pandas(
         if data_label.upper() in labeler.label_mapping.keys():
             data_label_ind = labeler.label_mapping[data_label.upper()]
         else:
-            raise ValueError(
+            raise ValueError(  # noqa: TRY003
                 """
                 The only values acceptable for the data label parameter are as follows:
                 ['PAD', 'UNKNOWN', 'ADDRESS', 'BAN', 'CREDIT_CARD', 'DATE', 'TIME', 'DATETIME',\

contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/metrics/data_profiler_metrics/data_profiler_profile_diff.py

@@ -30,7 +30,7 @@ def _pandas(
             first_profile_path = metric_value_kwargs["profile_path"]
             first_profile = dp.Profiler.load(first_profile_path)
         except FileNotFoundError:
-            raise ValueError(
+            raise ValueError(  # noqa: TRY003
                 "'profile_path' does not point to a valid DataProfiler stored profile."
             )
 

contrib/capitalone_dataprofiler_expectations/capitalone_dataprofiler_expectations/metrics/data_profiler_metrics/data_profiler_profile_report.py

@@ -29,9 +29,9 @@ def _pandas(
             profile_report["global_stats"]["profile_schema"] = dict(
                 profile_report["global_stats"]["profile_schema"]
             )
-            return profile_report
+            return profile_report  # noqa: TRY300
         except FileNotFoundError:
-            raise ValueError(
+            raise ValueError(  # noqa: TRY003
                 "'profile_path' does not point to a valid DataProfiler stored profile."
             )
         except Exception as e:

contrib/cli/great_expectations_contrib/package.py

@@ -251,7 +251,7 @@ def retrieve_package_expectations_diagnostics() -> List[ExpectationDiagnostics]:
                 )
             )
             diagnostics = GreatExpectationsContribPackageManifest._gather_diagnostics(expectations)
-            return diagnostics
+            return diagnostics  # noqa: TRY300
         except Exception as e:
             # Exceptions should not break the CLI - this behavior should be working in the background
             # without the user being concerned about the underlying functionality
@@ -271,9 +271,9 @@ def _identify_user_package() -> str:
 
         # A sanity check in case the user modifies the Cookiecutter template in unexpected ways
         if len(packages) == 0:
-            raise FileNotFoundError("Could not find a user-defined package")
+            raise FileNotFoundError("Could not find a user-defined package")  # noqa: TRY003
         elif len(packages) > 1:
-            raise ValueError("Found more than one user-defined package")
+            raise ValueError("Found more than one user-defined package")  # noqa: TRY003
 
         return packages[0]
 
@@ -284,8 +284,8 @@ def _import_expectations_module(package: str) -> Any:
         sys.path.append(cwd)
         try:
             expectations_module = importlib.import_module(f"{package}.expectations")
-            return expectations_module
-        except ModuleNotFoundError:
+            return expectations_module  # noqa: TRY300
+        except ModuleNotFoundError:  # noqa: TRY302
             raise
 
     @staticmethod

contrib/experimental/great_expectations_experimental/expectations/expect_column_distinct_values_to_be_continuous.py

@@ -246,7 +246,7 @@ def _expected_list(
                     for x in range((end_date - start_date).days + 1)
                 ]
             except TypeError as ex:
-                raise InvalidExpectationConfigurationError(
+                raise InvalidExpectationConfigurationError(  # noqa: TRY003
                     f"Expecting datetime when datetime_format is set\n{ex}"
                 )
         # else - no datetime format, so we're assuming integers

contrib/experimental/great_expectations_experimental/expectations/expect_column_skew_to_be_between.py

@@ -106,13 +106,13 @@ def _get_query_result(func, selectable, execution_engine: SqlAlchemyExecutionEng
 
     try:
         result: sqlalchemy.Row = execution_engine.execute_query(simple_query).fetchone()[0]
-        return result
+        return result  # noqa: TRY300
     except sqlalchemy.ProgrammingError as pe:
         exception_message: str = "An SQL syntax Exception occurred."
         exception_traceback: str = traceback.format_exc()
         exception_message += f'{type(pe).__name__}: "{pe!s}".  Traceback: "{exception_traceback}".'
-        logger.error(exception_message)
-        raise pe
+        logger.error(exception_message)  # noqa: TRY400
+        raise pe  # noqa: TRY201
 
     # @classmethod
     # def _get_evaluation_dependencies(

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_be_edtf_parseable.py

@@ -52,7 +52,7 @@ def _pandas(cls, column, level=None, **kwargs):
         def is_parseable(val):
             try:
                 if type(val) != str:  # noqa: E721
-                    raise TypeError(
+                    raise TypeError(  # noqa: TRY003
                         "Values passed to expect_column_values_to_be_edtf_parseable must be of type string.\nIf you want to validate a column of dates or timestamps, please call the expectation before converting from string format."
                     )
 
@@ -62,7 +62,7 @@ def is_parseable(val):
                 return False
 
         if level is not None and type(level) != int:  # noqa: E721
-            raise TypeError("level must be of type int.")
+            raise TypeError("level must be of type int.")  # noqa: TRY003
 
         return column.map(is_parseable)
 

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_be_in_set_spark_optimized.py

@@ -46,7 +46,7 @@ def _spark(
                 data=value_set, schema=df.schema[column_name].dataType
             )
         except TypeError as e:
-            raise InvalidExpectationConfigurationError(
+            raise InvalidExpectationConfigurationError(  # noqa: TRY003
                 f"`value_set` must be of same type as `column` : {e}"
             )
 

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_be_string_integers_increasing.py

@@ -44,7 +44,7 @@ def _pandas(self, _column, **kwargs):
         if all(_column.str.isdigit()) is True:
             temp_column = _column.astype(int)
         else:
-            raise TypeError("Column must be a string-type capable of being cast to int.")
+            raise TypeError("Column must be a string-type capable of being cast to int.")  # noqa: TRY003
 
         series_diff = np.diff(temp_column)
 
@@ -75,7 +75,7 @@ def _spark(
         if pyspark.types and isinstance(column_metadata["type"], pyspark.types.StringType):
             column = F.col(column_name).cast(pyspark.types.IntegerType())
         else:
-            raise TypeError("Column must be a string-type capable of being cast to int.")
+            raise TypeError("Column must be a string-type capable of being cast to int.")  # noqa: TRY003
 
         compute_domain_kwargs = metric_domain_kwargs
 
@@ -88,7 +88,7 @@ def _spark(
         )
 
         if any(np.array(df.select(column.isNull()).collect())):
-            raise TypeError("Column must be a string-type capable of being cast to int.")
+            raise TypeError("Column must be a string-type capable of being cast to int.")  # noqa: TRY003
 
         diff = column - F.lag(column).over(pyspark.Window.orderBy(F.lit("constant")))
         diff = F.when(diff.isNull(), 1).otherwise(diff)
@@ -196,7 +196,7 @@ def _validate_success_key(
         """"""
         if param not in configuration.kwargs:
             if required:
-                raise InvalidExpectationKwargsError(
+                raise InvalidExpectationKwargsError(  # noqa: TRY003
                     f"Parameter {param} is required but was not found in configuration."
                 )
             return

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_be_valid_github_username.py

@@ -25,7 +25,7 @@ def valid_github_user(x):
                 return False
             try:
                 urlopen(Request(BASE_API_URL + x, headers={"User-Agent": "great_expectations"}))
-                return True
+                return True  # noqa: TRY300
             except (URLError, HTTPError):
                 return False
 

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_be_valid_iata_code.py

@@ -27,7 +27,7 @@ def valid_iata_code(x):
                 return False
             try:
                 urlopen(Request(BASE_API_URL + x, headers={"User-Agent": "great_expectations"}))
-                return True
+                return True  # noqa: TRY300
             except (URLError, HTTPError):
                 return False
 

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_be_valid_json.py

@@ -33,7 +33,7 @@ def is_valid_json(x):
                 return False
             try:
                 loads(x)
-                return True
+                return True  # noqa: TRY300
             except ValueError:
                 return False
 

contrib/experimental/great_expectations_experimental/expectations/expect_column_values_to_change_between.py

@@ -30,7 +30,7 @@ class ColumnValuesToChangeBetween(ColumnMapMetricProvider):
     def _pandas(cls, column, from_value, to_value, **kwargs):
         # throw an error if one of the values is not numeric
         if not pd.to_numeric(column, errors="coerce").notnull().all():
-            raise TypeError("Column values must be numeric !")
+            raise TypeError("Column values must be numeric !")  # noqa: TRY003
 
         # calculate the difference of the current row with the previous.
         # If previous is NaN fills with the initial value "from_value" to consider it true