Update slither/tools/flattening/flattening.py

Co-authored-by: alpharush <0xalpharush@protonmail.com>
2 years ago · c1ed068ae1
parent e64de7f7b3
commit c1ed068ae1
1 changed files with 1 additions and 5 deletions
--- a/slither/tools/flattening/flattening.py
+++ b/slither/tools/flattening/flattening.py
@ -151,12 +151,8 @@ class Flattening:
        if self._convert_library_to_internal and contract.is_library:
            for f in contract.functions_declared:
                visibility = ""
-                if f.visibility == "external":
-                    visibility = f.visibility
-                elif f.visibility == "public":
+                if f.visibility in ["external", "public"]:
                    visibility = f.visibility
-
-                if visibility != "":
                    attributes_start = (
                        f.parameters_src().source_mapping["start"]
                        + f.parameters_src().source_mapping["length"]