Skip to content

Commit 11abfda

Browse files
refactoring
1 parent 48a7556 commit 11abfda

31 files changed

Lines changed: 38 additions & 93 deletions

code2logic/adaptive.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,8 @@
1313
result = reproducer.reproduce("path/to/file.py")
1414
"""
1515

16-
import os
17-
import re
1816
from pathlib import Path
19-
from typing import Dict, Any, List, Optional, Tuple
17+
from typing import Dict, Any, List
2018
from dataclasses import dataclass
2119

2220
# Load .env file

code2logic/analyzer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from pathlib import Path
99
from datetime import datetime
1010
from collections import defaultdict
11-
from typing import Optional, List, Dict
11+
from typing import List, Dict
1212

1313
from .models import ProjectInfo, ModuleInfo
1414
from .parsers import TreeSitterParser, UniversalParser, TREE_SITTER_AVAILABLE

code2logic/base.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
"""
99

1010
import logging
11-
from typing import Optional
1211

1312

1413
class VerboseMixin:

code2logic/benchmark.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
results = benchmark.run_all("path/to/file.py")
1616
"""
1717

18-
import os
1918
import json
2019
import time
2120

@@ -33,7 +32,6 @@
3332
from .analyzer import analyze_project
3433
from .generators import (
3534
MarkdownGenerator,
36-
CompactGenerator,
3735
JSONGenerator,
3836
YAMLGenerator,
3937
CSVGenerator,

code2logic/benchmarks/runner.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -15,28 +15,24 @@
1515
result.save('output/benchmark.json')
1616
"""
1717

18-
import re
1918
import sys
2019
import time
2120
from pathlib import Path
22-
from typing import Dict, List, Optional, Tuple, Callable
23-
from concurrent.futures import ThreadPoolExecutor, as_completed
24-
from dataclasses import dataclass
21+
from typing import List, Optional
2522

2623
from ..analyzer import analyze_project
2724
from ..llm_clients import get_client, BaseLLMClient
28-
from ..reproduction import extract_code_block
2925
from ..utils import estimate_tokens
3026
from ..metrics import ReproductionMetrics
31-
from ..terminal import render, ShellRenderer
27+
from ..terminal import render
3228

3329
from .results import (
3430
BenchmarkResult, BenchmarkConfig, FileResult,
3531
FunctionResult, FormatResult
3632
)
3733
from .common import (
38-
generate_spec, generate_spec_token, create_single_project,
39-
get_token_reproduction_prompt, get_simple_reproduction_prompt
34+
generate_spec_token, create_single_project,
35+
get_token_reproduction_prompt
4036
)
4137

4238

@@ -605,7 +601,6 @@ def _test_function(
605601
result.similarity = SequenceMatcher(None, orig_norm, repr_norm).ratio() * 100
606602

607603
if verbose:
608-
status = "✓" if result.similarity > 50 else "○"
609604
syntax = "S✓" if result.syntax_ok else "S✗"
610605
print(f" {func.name}: {result.similarity:.1f}% {syntax}")
611606

code2logic/chunked_reproduction.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,9 @@
1212
"""
1313

1414
import re
15-
from dataclasses import dataclass, field
16-
from typing import Dict, List, Optional, Tuple
17-
from pathlib import Path
15+
from dataclasses import dataclass
16+
from typing import List
1817

19-
from .models import ProjectInfo, ModuleInfo, FunctionInfo, ClassInfo
2018
from .utils import estimate_tokens
2119

2220

code2logic/cli.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
import sys
1414
import subprocess
1515
import time
16-
import logging
1716
import json
1817
import signal
1918
from datetime import datetime
@@ -719,9 +718,8 @@ def _maybe_print_pretty_help() -> bool:
719718
from .analyzer import ProjectAnalyzer, get_library_status
720719
from .generators import (
721720
MarkdownGenerator, CompactGenerator, JSONGenerator,
722-
YAMLGenerator, CSVGenerator
721+
YAMLGenerator
723722
)
724-
from .gherkin import GherkinGenerator
725723
from .toon_format import TOONGenerator
726724
from .logicml import LogicMLGenerator
727725
from .function_logic import FunctionLogicGenerator

code2logic/dependency.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
"""
77

88
from pathlib import Path
9-
from typing import Dict, List, Optional
9+
from typing import Dict, List
1010

1111
from .models import ModuleInfo, DependencyNode
1212

code2logic/errors.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from pathlib import Path
1010
from typing import List, Dict, Any, Optional, Callable
1111
import logging
12-
import traceback
1312

1413

1514
class ErrorSeverity(Enum):

code2logic/file_formats.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,9 @@
55
optimized for LLM code reproduction.
66
"""
77

8-
import re
98
import json
109
from pathlib import Path
11-
from typing import Dict, Any, List
10+
from typing import Dict, Any
1211

1312

1413
def generate_file_csv(file_path: Path) -> str:

0 commit comments

Comments
 (0)