-
Notifications
You must be signed in to change notification settings - Fork 24
Open
Description
I'm using lief version 0.12.3 and line 412 throws and exception.
It needs to be modifed to: "has_signature": int(lief_binary.has_signatures),.
I can send a PR :)
Traceback (most recent call last):
File "ppo.py", line 30, in <module>
agent.learn(total_timesteps=2500)
File "/home/mari/miniconda3/envs/malware_rl/lib/python3.8/site-packages/stable_baselines3/ppo/ppo.py", line 299, in learn
return super(PPO, self).learn(
File "/home/mari/miniconda3/envs/malware_rl/lib/python3.8/site-packages/stable_baselines3/common/on_policy_algorithm.py", line 242, in learn
total_timesteps, callback = self._setup_learn(
File "/home/mari/miniconda3/envs/malware_rl/lib/python3.8/site-packages/stable_baselines3/common/base_class.py", line 429, in _setup_learn
self._last_obs = self.env.reset() # pytype: disable=annotation-type-mismatch
File "/home/mari/miniconda3/envs/malware_rl/lib/python3.8/site-packages/stable_baselines3/common/vec_env/dummy_vec_env.py", line 61, in reset
obs = self.envs[env_idx].reset()
File "/home/mari/miniconda3/envs/malware_rl/lib/python3.8/site-packages/stable_baselines3/common/monitor.py", line 79, in reset
return self.env.reset(**kwargs)
File "/home/mari/miniconda3/envs/malware_rl/lib/python3.8/site-packages/gym/wrappers/monitor.py", line 38, in reset
observation = self.env.reset(**kwargs)
File "/home/mari/repos/malware_rl/malware_rl/envs/sorel_gym.py", line 128, in reset
self.observation_space = self.feature_extractor(self.bytez)
File "/home/mari/repos/malware_rl/malware_rl/envs/utils/sorel.py", line 24, in extract
return np.array(self.extractor.feature_vector(bytez), dtype=np.float32)
File "/home/mari/repos/malware_rl/malware_rl/envs/utils/ember.py", line 764, in feature_vector
return self.process_raw_features(self.raw_features(bytez))
File "/home/mari/repos/malware_rl/malware_rl/envs/utils/ember.py", line 750, in raw_features
{fe.name: fe.raw_features(bytez, lief_binary) for fe in self.features},
File "/home/mari/repos/malware_rl/malware_rl/envs/utils/ember.py", line 750, in <dictcomp>
{fe.name: fe.raw_features(bytez, lief_binary) for fe in self.features},
File "/home/mari/repos/malware_rl/malware_rl/envs/utils/ember.py", line 412, in raw_features
"has_signature": int(lief_binary.has_signature),
AttributeError: 'lief.PE.Binary' object has no attribute 'has_signature'
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels