lint

Author: tomvanmele

src/compas_fea2/base.py

@@ -135,6 +135,7 @@ def from_name(cls, name, **kwargs):
     # ==========================================================================
     # Copy and Serialization
     # ==========================================================================
+
     def copy(self, cls=None, copy_guid=False, copy_name=False):
         """Make an independent copy of the data object.
 

src/compas_fea2/job/input_file.py

@@ -73,8 +73,7 @@ def write_to_file(self, path=None):
 
 
 class ParametersFile(InputFile):
-    """Input file object for Optimizations.
-    """
+    """Input file object for Optimizations."""
 
     def __init__(self, **kwargs):
         super(ParametersFile, self).__init__(**kwargs)

src/compas_fea2/model/elements.py

@@ -347,8 +347,8 @@ def plot_section(self):
         self.section.plot()
 
     def plot_stress_distribution(self, step: "_Step", end: str = "end_1", nx: int = 100, ny: int = 100, *args, **kwargs):  # noqa: F821
-        """ Plot the stress distribution along the element.
-        
+        """Plot the stress distribution along the element.
+
         Parameters
         ----------
         step : :class:`compas_fea2.model.Step`
@@ -375,41 +375,41 @@ def section_forces_result(self, step: "Step") -> "Result":  # noqa: F821
         ----------
         step : :class:`compas_fea2.model.Step`
             The analysis step.
-        
+
         Returns
         -------
         :class:`compas_fea2.results.Result`
             The section forces result for the element.
         """
-        
+
         if not hasattr(step, "section_forces_field"):
             raise ValueError("The step does not have a section_forces_field")
         return step.section_forces_field.get_result_at(self)
 
     def forces(self, step: "Step") -> "Result":  # noqa: F821
         """Get the forces result for the element.
-        
+
         Parameters
         ----------
         step : :class:`compas_fea2.model.Step`
             The analysis step.
-        
+
         Returns
         -------
         :class:`compas_fea2.results.Result`
-            The forces result for the element.     
+            The forces result for the element.
         """
         r = self.section_forces_result(step)
         return r.forces
 
     def moments(self, step: "_Step") -> "Result":  # noqa: F821
-        """ Get the moments result for the element.
-        
+        """Get the moments result for the element.
+
         Parameters
         ----------
         step : :class:`compas_fea2.model.Step`
             The analysis step.
-            
+
         Returns
         -------
         :class:`compas_fea2.results.Result`
@@ -638,12 +638,12 @@ def _construct_faces(self, face_indices: Dict[str, Tuple[int]]) -> List[Face]:
 
     def stress_results(self, step: "_Step") -> "Result":  # noqa: F821
         """Get the stress results for the element.
-        
+
         Parameters
         ----------
         step : :class:`compas_fea2.model.Step`
             The analysis step.
-            
+
         Returns
         -------
         :class:`compas_fea2.results.Result`

src/compas_fea2/model/nodes.py

@@ -265,7 +265,7 @@ def connected_elements(self) -> List:
 
     def transform(self, transformation):
         """Transform the node using a transformation matrix.
-        
+
         Parameters
         ----------
         transformation : list
@@ -275,12 +275,12 @@ def transform(self, transformation):
 
     def transformed(self, transformation):
         """Return a copy of the node transformed by a transformation matrix.
-        
+
         Parameters
         ----------
         transformation : list
             A 4x4 transformation matrix.
-            
+
         Returns
         -------
         :class:`compas_fea2.model.Node`
@@ -292,12 +292,12 @@ def transformed(self, transformation):
 
     def displacement(self, step):
         """Get the displacement of the node at a given step.
-        
+
         Parameters
         ----------
         step : :class:`compas_fea2.model.Step`
             The step for which to get the displacement.
-            
+
         Returns
         -------
         :class:`compas_fea2.results.DisplacementResult`
@@ -308,18 +308,18 @@ def displacement(self, step):
 
     def reaction(self, step):
         """Get the reaction of the node at a given step.
-        
+
         Parameters
         ----------
         step : :class:`compas_fea2.model.Step`
             The step for which to get the reaction.
-        
+
         Returns
         -------
         :class:`compas_fea2.results.ReactionResult`
             The reaction result at the node for the given step.
         """
-        
+
         if step.reaction_field:
             return step.reaction_field.get_result_at(location=self)
 

src/compas_fea2/model/parts.py

@@ -598,7 +598,7 @@ def elements_by_dimension(self, dimension: int = 1) -> Iterable[_Element]:
         Iterable[:class:`compas_fea2.model._Element`]
             The elements of the specified dimension.
         """
-        
+
         if dimension == 1:
             return filter(lambda x: isinstance(x, _Element1D), self.elements)
         elif dimension == 2:
@@ -942,7 +942,7 @@ def from_step_file(cls, step_file: str, name: Optional[str] = None, **kwargs) ->
     def from_brep(cls, brep, name: Optional[str] = None, **kwargs) -> "_Part":
         """Create a Part object from a BREP file.
         Parameters
-        ---------- 
+        ----------
         brep : str
             Path to the BREP file.
         name : str, optional
@@ -1551,8 +1551,8 @@ def visualize_node_connectivity(self):
 
     def visualize_pyvis(self, filename="model_graph.html"):
         """Visualizes the Model-Part and Element-Node graph using Pyvis.
-        The graph is saved as an HTML file, which can be opened in a web browser.  
-        
+        The graph is saved as an HTML file, which can be opened in a web browser.
+
         Warnings
         --------
         The Pyvis library must be installed to use this function. This function

src/compas_fea2/model/releases.py

@@ -1,5 +1,10 @@
+from typing import TYPE_CHECKING
+
 from compas_fea2.base import FEAData
 
+if TYPE_CHECKING:
+    from .elements import BeamElement
+
 
 class _BeamEndRelease(FEAData):
     """Assign a general end release to a `compas_fea2.model.BeamElement`.
@@ -40,9 +45,18 @@ class _BeamEndRelease(FEAData):
 
     """
 
-    def __init__(self, n: bool = False, v1: bool = False, v2: bool = False, m1: bool = False, m2: bool = False, t: bool = False, **kwargs):
+    def __init__(
+        self,
+        n: bool = False,
+        v1: bool = False,
+        v2: bool = False,
+        m1: bool = False,
+        m2: bool = False,
+        t: bool = False,
+        **kwargs,
+    ):
         super().__init__(**kwargs)
-        self._element: "BeamElement | None"  # type: ignore
+        self._element: "BeamElement | None" = None
         self._location: str | None = None
         self.n: bool = n
         self.v1: bool = v1
@@ -52,12 +66,12 @@ def __init__(self, n: bool = False, v1: bool = False, v2: bool = False, m1: bool
         self.t: bool = t
 
     @property
-    def element(self) -> "BeamElement | None":  # type: ignore
+    def element(self) -> "BeamElement | None":
         return self._element
 
     @element.setter
-    def element(self, value: "BeamElement"):  # type: ignore
-        if not isinstance(value, "BeamElement"):  # type: ignore
+    def element(self, value: "BeamElement"):
+        if not isinstance(value, "BeamElement"):
             raise TypeError(f"{value!r} is not a beam element.")
         self._element = value
 

src/compas_fea2/problem/combinations.py

@@ -1,5 +1,6 @@
-from compas_fea2.base import FEAData
 import compas_fea2
+from compas_fea2.base import FEAData
+
 
 class LoadCombination(FEAData):
     """Load combination used to combine load fields together at each step.
@@ -68,7 +69,7 @@ def node_load(self):
         for load_field in self.step.load_fields:
             if isinstance(load_field, compas_fea2.problem.LoadField):
                 if load_field.load_case in self.factors:
-                    for node in load_field.distribution :
+                    for node in load_field.distribution:
                         for load in load_field.loads:
                             if node in nodes_loads:
                                 nodes_loads[node] += load * self.factors[load_field.load_case]

src/compas_fea2/problem/steps/step.py

@@ -602,14 +602,14 @@ def add_gravity_load(self, parts=None, g=9.81, x=0.0, y=0.0, z=-1.0, load_case=N
 
     def add_temperature_field(self, field, node):
         """Add a temperature field to the Step object.
-        
+
         Parameters
         ----------
         field : :class:`compas_fea2.problem.fields.PrescribedTemperatureField`
             The temperature field to add.
         node : :class:`compas_fea2.model.Node`
             The node to which the temperature field is applied.
-            
+
         Returns
         -------
         :class:`compas_fea2.problem.fields.PrescribedTemperatureField`
@@ -621,7 +621,7 @@ def add_temperature_field(self, field, node):
 
         # if not isinstance(node, Node):
         #     raise TypeError("{!r} is not a Node.".format(node))
-    
+
         # node._temperature = field
         # self._fields.setdefault(node.part, {}).setdefault(field, set()).add(node)
         # return field