0d66931e6da93fda1c07388e6a2ed16d9a129040,src/syft/core/store/store_memory.py,MemoryStore,get_objects_of_type,#MemoryStore#Any#,33

Before Change

        self.post_init()

    def get_objects_of_type(self, obj_type: type) -> Set[AbstractStorableObject]:
        results = set()
        for key, obj in self._objects.items():
            if isinstance(obj.data, obj_type):
                results.add(obj)

        return results

    @syft_decorator(typechecking=True)
    def __sizeof__(self) -> int:
        return self._objects.__sizeof__()

After Change

        self.post_init()

    def get_objects_of_type(self, obj_type: type) -> Set[AbstractStorableObject]:
        return {obj for obj in self.values() if isinstance(obj.data, obj_type)}

    @syft_decorator(typechecking=True)
    def __sizeof__(self) -> int:
        return self._objects.__sizeof__()

Instances