Skip to content

Commit 09ad916

Browse files
authored
JVM-introspector: Remove redundant FI call (#793)
This PR remove redundant FI calls. --------- Signed-off-by: Arthur Chan <[email protected]>
1 parent 13fbcb1 commit 09ad916

File tree

2 files changed

+2
-12
lines changed

2 files changed

+2
-12
lines changed

data_prep/introspector.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,6 @@ def get_oracle_dict() -> Dict[str, Any]:
8787
'far-reach-low-coverage': get_unreached_functions,
8888
'low-cov-with-fuzz-keyword': query_introspector_for_keyword_targets,
8989
'easy-params-far-reach': query_introspector_for_easy_param_targets,
90-
'jvm-public-candidates': query_introspector_jvm_all_public_candidates,
9190
'optimal-targets': query_introspector_for_optimal_targets,
9291
'test-migration': query_introspector_for_tests,
9392
'all-public-candidates': query_introspector_all_public_candidates,
@@ -274,19 +273,10 @@ def query_introspector_for_easy_param_targets(project: str) -> list[dict]:
274273
return query_introspector_oracle(project, INTROSPECTOR_ORACLE_EASY_PARAMS)
275274

276275

277-
def query_introspector_jvm_all_public_candidates(project: str) -> list[dict]:
278-
"""Queries Fuzz Introspector for all public accessible function or
279-
constructor candidates.
280-
"""
281-
return query_introspector_oracle(project,
282-
INTROSPECTOR_ORACLE_ALL_PUBLIC_CANDIDATES)
283-
284-
285276
def query_introspector_all_public_candidates(project: str) -> list[dict]:
286277
"""Queries Fuzz Introspector for all public accessible function or
287278
constructor candidates.
288279
"""
289-
#TODO May combine this with query_introspector_jvm_all_public_candidates
290280
return query_introspector_oracle(project,
291281
INTROSPECTOR_ORACLE_ALL_PUBLIC_CANDIDATES)
292282

llm_toolkit/prompt_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1225,7 +1225,7 @@ def build(self,
12251225
'\n---\n'.join(source_list))
12261226

12271227
# Add all public candidates to prompt
1228-
methods = introspector.query_introspector_jvm_all_public_candidates(proj)
1228+
methods = introspector.query_introspector_all_public_candidates(proj)
12291229
name = [method['function_name'] for method in methods]
12301230
prompt_text = prompt_text.replace('{PUBLIC_METHODS}', ','.join(name))
12311231
else:
@@ -1585,7 +1585,7 @@ def _extract_jvm_imports(self, src: str, cls: list[str]) -> list[str]:
15851585
def _get_jvm_public_candidates(self, proj: str) -> list[str]:
15861586
"""Helper function to retrieve list of public candidates for jvm."""
15871587
method_set = set()
1588-
methods = introspector.query_introspector_jvm_all_public_candidates(proj)
1588+
methods = introspector.query_introspector_all_public_candidates(proj)
15891589
for method in methods:
15901590
if "<init>" not in method['function_name']:
15911591
method_set.add(method['function_name'])

0 commit comments

Comments
 (0)