From b06f21cdc85d5731fff3897a5ba67c8cf61108d7 Mon Sep 17 00:00:00 2001
From: Oscar Gustafsson <oscar.gustafsson@gmail.com>
Date: Fri, 17 Feb 2023 23:01:54 +0100
Subject: [PATCH] Import from correct location
---
b_asic/signal.py | 26 ++++++--------------------
1 file changed, 6 insertions(+), 20 deletions(-)
diff --git a/b_asic/signal.py b/b_asic/signal.py
index f2ce69bb..8b094a94 100644
--- a/b_asic/signal.py
+++ b/b_asic/signal.py
@@ -5,12 +5,8 @@ Contains the class for representing the connections between operations.
"""
from typing import TYPE_CHECKING, Iterable, Optional, Union
-from b_asic.graph_component import (
- AbstractGraphComponent,
- GraphComponent,
- Name,
- TypeName,
-)
+from b_asic.graph_component import AbstractGraphComponent, GraphComponent
+from b_asic.types import Name, TypeName
if TYPE_CHECKING:
from b_asic.operation import Operation
@@ -48,9 +44,7 @@ class Signal(AbstractGraphComponent):
def __init__(
self,
source: Optional[Union["OutputPort", "Signal", "Operation"]] = None,
- destination: Optional[
- Union["InputPort", "Signal", "Operation"]
- ] = None,
+ destination: Optional[Union["InputPort", "Signal", "Operation"]] = None,
bits: Optional[int] = None,
name: Name = Name(""),
):
@@ -70,11 +64,7 @@ class Signal(AbstractGraphComponent):
@property
def neighbors(self) -> Iterable[GraphComponent]:
- return [
- p.operation
- for p in [self.source, self.destination]
- if p is not None
- ]
+ return [p.operation for p in [self.source, self.destination] if p is not None]
@property
def source(self) -> Optional["OutputPort"]:
@@ -86,9 +76,7 @@ class Signal(AbstractGraphComponent):
"""The destination InputPort of the signal."""
return self._destination
- def set_source(
- self, source: Union["OutputPort", "Signal", "Operation"]
- ) -> None:
+ def set_source(self, source: Union["OutputPort", "Signal", "Operation"]) -> None:
"""
Disconnect the previous source OutputPort of the signal and
connect to the entered source OutputPort. Also connect the entered
@@ -197,9 +185,7 @@ class Signal(AbstractGraphComponent):
"""
if bits is not None:
if not isinstance(bits, int):
- raise TypeError(
- f"Bits must be an int, not {type(bits)}: {bits!r}"
- )
+ raise TypeError(f"Bits must be an int, not {type(bits)}: {bits!r}")
if bits < 0:
raise ValueError("Bits cannot be negative")
self.set_param("bits", bits)
--
GitLab