Skip to content

Commit 2f62370

Browse files
committed
Temp fix 6.
1 parent fa5576b commit 2f62370

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

fusil/python/write_python_code.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -550,6 +550,8 @@ def _write_main_fuzzing_logic(self) -> None:
550550
if not self.module_classes:
551551
break
552552
class_name = choice(self.module_classes)
553+
if class_name in OBJECT_BLACKLIST:
554+
continue
553555
try:
554556
class_obj = getattr(self.module, class_name)
555557
except AttributeError:
@@ -588,7 +590,11 @@ def _fuzz_one_class(self, class_idx: int, class_name_str: str, class_type: type)
588590
self.write_print_to_stderr(
589591
0, f'"[{prefix}] Attempting to instantiate class: {class_name_str}"'
590592
)
591-
593+
if class_name_str in OBJECT_BLACKLIST:
594+
self.write_print_to_stderr(
595+
0, f'"[{prefix}] Skipping blacklisted class: {class_name_str}"'
596+
)
597+
return
592598
instance_var_name = (
593599
f"instance_{prefix}_{class_name_str.lower().replace('.', '_')}" # Unique name
594600
)

0 commit comments

Comments
 (0)