diff options
Diffstat (limited to 'benchmark/agbenchmark/__main__.py')
-rw-r--r-- | benchmark/agbenchmark/__main__.py | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/benchmark/agbenchmark/__main__.py b/benchmark/agbenchmark/__main__.py index a111dd9ff..f6c28d8eb 100644 --- a/benchmark/agbenchmark/__main__.py +++ b/benchmark/agbenchmark/__main__.py @@ -14,15 +14,15 @@ from agbenchmark.utils.logging import configure_logging load_dotenv() -try: - if os.getenv("HELICONE_API_KEY"): - import helicone # noqa +# try: +# if os.getenv("HELICONE_API_KEY"): +# import helicone # noqa - helicone_enabled = True - else: - helicone_enabled = False -except ImportError: - helicone_enabled = False +# helicone_enabled = True +# else: +# helicone_enabled = False +# except ImportError: +# helicone_enabled = False class InvalidInvocationError(ValueError): @@ -35,12 +35,12 @@ BENCHMARK_START_TIME_DT = datetime.now(timezone.utc) BENCHMARK_START_TIME = BENCHMARK_START_TIME_DT.strftime("%Y-%m-%dT%H:%M:%S+00:00") -if helicone_enabled: - from helicone.lock import HeliconeLockManager +# if helicone_enabled: +# from helicone.lock import HeliconeLockManager - HeliconeLockManager.write_custom_property( - "benchmark_start_time", BENCHMARK_START_TIME - ) +# HeliconeLockManager.write_custom_property( +# "benchmark_start_time", BENCHMARK_START_TIME +# ) @click.group(cls=DefaultGroup, default_if_no_args=True) |