Skip to content

Commit

Permalink
[Bug Fix] Use activeCodeTypes() instead of codeTypes()
Browse files Browse the repository at this point in the history
Used the wrong function when handled a single active code type. Fixes
issue CheckPointSW#41.
  • Loading branch information
chkp-eyalit committed May 28, 2020
1 parent 0603cb8 commit 2b5dfe1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/thumbs_up/utils/function.py
Original file line number Diff line number Diff line change
Expand Up @@ -410,7 +410,7 @@ def predictFunctionStartType(self, ea):
"""
# Nothing to check if there is only one type
if not self._analyzer.hasActiveCodeTypes():
return self._analyzer.codeTypes()[0]
return self._analyzer.activeCodeTypes()[0]
# Multiple types, now predict the right one
sample = self.extractFunctionTypeSample(ea)
return int(self._type_classifier.predict([sample]))
Expand Down

0 comments on commit 2b5dfe1

Please sign in to comment.