From f4397bf940e3a54a745ac1be19cadb9e33a28ae4 Mon Sep 17 00:00:00 2001 From: Rita Aleksziev Date: Wed, 8 Jan 2025 12:33:14 +0100 Subject: [PATCH] Remove setting envvars from arg --- examples/python/code_graph_example.py | 22 ---------------------- 1 file changed, 22 deletions(-) diff --git a/examples/python/code_graph_example.py b/examples/python/code_graph_example.py index afc83beb0..16eafb024 100644 --- a/examples/python/code_graph_example.py +++ b/examples/python/code_graph_example.py @@ -24,18 +24,6 @@ def parse_args(): default=True, help="Whether or not to process non-code files" ) - parser.add_argument( - "--mock_embedding", - type=lambda x: x.lower() in ("true", "1"), - default=True, - help="Whether or not to mock embedding and code summary" - ) - parser.add_argument( - "--mock_code_summary", - type=lambda x: x.lower() in ("true", "1"), - default=True, - help="Whether or not to mock code summary" - ) parser.add_argument( "--time", type=lambda x: x.lower() in ("true", "1"), @@ -47,18 +35,8 @@ def parse_args(): if __name__ == "__main__": setup_logging(logging.ERROR) - import os - args = parse_args() - if args.mock_embedding: - os.environ["MOCK_EMBEDDING"] = "true" - print("Mocking embedding.") - - if args.mock_code_summary: - os.environ["MOCK_CODE_SUMMARY"] = "true" - print("Mocking code summary.") - if args.time: import time start_time = time.time()