Update airbyte/sources/util.py

airbytehq · Oct 6, 2024 · eb9b45a · eb9b45a
1 parent f5ef21e
commit eb9b45a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airbyte/sources/util.py b/airbyte/sources/util.py
@@ -140,7 +140,7 @@ def get_benchmark_source(
     if isinstance(num_records, str):
         try:
             num_records = int(Decimal(num_records.replace("_", "")))
-        except ValueError as ex:
+        except InvalidOperation as ex:
             raise PyAirbyteInputError(
                 message="Invalid number format.",
                 original_exception=ex,