Add support for removing empty resources from architecture

b_asic/architecture.py

@@ -764,19 +764,55 @@ of :class:`~b_asic.architecture.ProcessingElement`
                     d_out[i][v] += 1
         return [dict(d) for d in d_in], [dict(d) for d in d_out]
 
-    def resource_from_name(self, name: str):
+    def resource_from_name(self, name: str) -> Resource:
+        """
+        Get :class:`Resource` based on name.
+
+        Parameters
+        ----------
+        name : str
+            Name of the resource.
+
+        Returns
+        -------
+        :class:`Resource`
+
+        """
         re = {p.entity_name: p for p in chain(self.memories, self.processing_elements)}
         return re[name]
 
+    def remove_resource(
+        self,
+        resource: Union[str, Resource],
+    ) -> None:
+        """
+        Remove an empty :class:`Resource` from the architecture.
+
+        Parameters
+        ----------
+        resource : :class:`b_asic.architecture.Resource` or str
+            The resource or the resource name to remove.
+        """
+        if isinstance(resource, str):
+            resource = self.resource_from_name(resource)
+
+        if resource.collection:
+            raise ValueError("Resource must be empty")
+
+        if resource in self.memories:
+            self.memories.remove(resource)
+        else:
+            self.processing_elements.remove(resource)
+
     def move_process(
         self,
         proc: Union[str, Process],
         re_from: Union[str, Resource],
         re_to: Union[str, Resource],
         assign: bool = False,
-    ):
+    ) -> None:
         """
-        Move a :class:`b_asic.process.Process` from one resource to another.
+        Move a :class:`b_asic.process.Process` from one :class:`Resource`  to another.
 
         Both the resource moved from and will become unassigned after a process has been
         moved, unless *assign* is set to True.

examples/fivepointwinograddft.py

@@ -196,7 +196,12 @@ memories[0].assign()
 memories[1].assign()
 memories[4].assign()
 
-for memory in memories:
+# %%
+# Memory 4 is now empty, so remove it.
+
+arch.remove_resource('memory4')
+
+for memory in arch.memories:
     memory.show_content(title=f"Improved {memory.entity_name}")
 
 arch