13f71c5eec807cbd8a368686fa7acd8d78be2303,networkml/featurizers/funcs/host.py,Host,_row_keys,#Host#Any#,577

Before Change

class Host(HostBase, Features):

    def _row_keys(self, row):
        keys = set()
        for field in ("eth.src", "eth.dst"):
            val = row.get(field, None)
            if not val:
                continue
            if not self._is_unicast(val):
                continue
            keys.add(val)
        return keys

    def pyshark_ipv4(self, rows):
        return self._pyshark_ipv4(rows)

After Change

class Host(HostBase, Features):

    def _row_keys(self, row):
        return {val for val in (row.get("eth.src", None), row.get("eth.dst", None)) if val and self._is_unicast(val)}

    def pyshark_ipv4(self, rows_f):
        return self._pyshark_ipv4(rows_f)

Instances