Merge pull request #347 from ie3-institute/df/#346-duplication

Removed duplicated code within `plots.grid`

Author: staudtMarius

CHANGELOG.md

@@ -10,6 +10,7 @@ All notable changes to this project will be documented in this file.
 ### Changed
 
 ### Removed
+- Removed duplicated code within `plots.grid` [#346](https://github.com/ie3-institute/pypsdm/issues/346)
 
 ### Fixed
 - Removed `q_characteristics` from `create_energy_management_systems_data` [#326](https://github.com/ie3-institute/pypsdm/issues/326)

pypsdm/plots/grid.py

@@ -191,13 +191,7 @@ def _node_trace(data, color):
 
     if highlights is not None:
         if isinstance(highlights, dict):
-            rmd = []
-            for nodes in highlights.values():
-                rmd.extend(nodes)
-            rmd = nodes_data.drop(rmd)
-            _node_trace(rmd, BLUE)
-
-            # plot highlighted nodes second so they are on top
+            # plot not highlighted nodes first
             rmd = []
             for nodes in highlights.values():
                 rmd.extend(nodes)
@@ -209,10 +203,7 @@ def _node_trace(data, color):
                 highlighted_nodes = nodes_data.loc[nodes]
                 _node_trace(highlighted_nodes, color)
         elif isinstance(highlights, list):
-            rmd = nodes_data.drop(highlights)
-            _node_trace(rmd, BLUE)
-
-            # plot highlighted nodes second so they are on top
+            # plot not highlighted nodes first
             rmd = nodes_data.drop(highlights)
             _node_trace(rmd, BLUE)