ruff check

Author: FBumann

flixopt/aggregation.py

@@ -27,8 +27,8 @@
 from .flow_system import FlowSystem
 from .structure import (
     Element,
-    Submodel,
     FlowSystemModel,
+    Submodel,
 )
 
 if TYPE_CHECKING:

flixopt/components.py

@@ -14,8 +14,8 @@
 from .elements import Component, ComponentModel, Flow
 from .features import InvestmentModel, OnOffModel, PiecewiseModel
 from .interface import InvestParameters, OnOffParameters, PiecewiseConversion
+from .modeling import BoundingPatterns
 from .structure import FlowSystemModel, register_class_for_io
-from.modeling import BoundingPatterns
 
 if TYPE_CHECKING:
     from .flow_system import FlowSystem

flixopt/effects.py

@@ -15,7 +15,7 @@
 
 from .core import Scalar, TemporalData, TemporalDataUser
 from .features import ShareAllocationModel
-from .structure import Element, ElementModel, Interface, Submodel, FlowSystemModel, register_class_for_io
+from .structure import Element, ElementModel, FlowSystemModel, Interface, Submodel, register_class_for_io
 
 if TYPE_CHECKING:
     from .flow_system import FlowSystem

flixopt/elements.py

@@ -13,10 +13,10 @@
 from .config import CONFIG
 from .core import PlausibilityError, Scalar, TemporalData, TemporalDataUser
 from .effects import TemporalEffectsUser
-from .features import InvestmentModel, OnOffModel, ModelingPrimitives
+from .features import InvestmentModel, ModelingPrimitives, OnOffModel
 from .interface import InvestParameters, OnOffParameters
-from .structure import Element, ElementModel, FlowSystemModel, register_class_for_io
 from .modeling import BoundingPatterns, ModelingUtilitiesAbstract
+from .structure import Element, ElementModel, FlowSystemModel, register_class_for_io
 
 if TYPE_CHECKING:
     from .flow_system import FlowSystem
@@ -626,7 +626,6 @@ def _do_modeling(self):
 
         if self.element.prevent_simultaneous_flows:
             # Simultanious Useage --> Only One FLow is On at a time, but needs a Binary for every flow
-            on_variables = [flow.submodel.on_off.on for flow in self.element.prevent_simultaneous_flows]
             ModelingPrimitives.mutual_exclusivity_constraint(
                 self,
                 binary_variables=[flow.submodel.on_off.on for flow in self.element.prevent_simultaneous_flows],

flixopt/features.py

@@ -10,10 +10,10 @@
 import numpy as np
 
 from .config import CONFIG
-from .core import NonTemporalData, Scalar, TemporalData, FlowSystemDimensions
+from .core import FlowSystemDimensions, NonTemporalData, Scalar, TemporalData
 from .interface import InvestParameters, OnOffParameters, Piecewise, PiecewiseEffects
-from .structure import Submodel, FlowSystemModel
-from .modeling import ModelingUtilities, ModelingPrimitives, BoundingPatterns
+from .modeling import BoundingPatterns, ModelingPrimitives, ModelingUtilities
+from .structure import FlowSystemModel, Submodel
 
 logger = logging.getLogger('flixopt')
 

flixopt/flow_system.py

@@ -32,7 +32,7 @@
     TemporalEffectsUser,
 )
 from .elements import Bus, Component, Flow
-from .structure import Element, Interface, FlowSystemModel
+from .structure import Element, FlowSystemModel, Interface
 
 if TYPE_CHECKING:
     import pyvis

flixopt/modeling.py

@@ -6,8 +6,8 @@
 import xarray as xr
 
 from .config import CONFIG
-from .core import NonTemporalData, Scalar, TemporalData, FlowSystemDimensions
-from .structure import Submodel, FlowSystemModel
+from .core import FlowSystemDimensions, NonTemporalData, Scalar, TemporalData
+from .structure import FlowSystemModel, Submodel
 
 logger = logging.getLogger('flixopt')
 

flixopt/structure.py

@@ -8,7 +8,7 @@
 import logging
 import pathlib
 from io import StringIO
-from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional, Tuple, Union, Collection
+from typing import TYPE_CHECKING, Any, Collection, Dict, List, Literal, Optional, Tuple, Union
 
 import linopy
 import numpy as np
@@ -19,7 +19,7 @@
 
 from . import io as fx_io
 from .config import CONFIG
-from .core import NonTemporalData, Scalar, TemporalDataUser, TimeSeriesData, get_dataarray_stats, FlowSystemDimensions
+from .core import FlowSystemDimensions, NonTemporalData, Scalar, TemporalDataUser, TimeSeriesData, get_dataarray_stats
 
 if TYPE_CHECKING:  # for type checking and preventing circular imports
     from .effects import EffectCollectionModel

tests/test_flow.py

@@ -641,7 +641,6 @@ def test_consecutive_on_hours_previous(self, basic_flow_system_linopy):
          'Sink(Wärme)|consecutive_on_hours|forward',
          'Sink(Wärme)|consecutive_on_hours|backward',
          'Sink(Wärme)|consecutive_on_hours|initial',
-         'Sink(Wärme)|consecutive_on_hours|lb',
         }.issubset(set(flow.submodel.constraints))
 
         assert_var_equal(