Skip to content
Snippets Groups Projects

Resolve "REFAC: Map Plot for more subsets"

Merged Ghost User requested to merge lukas_issue227_refac_map-plot-for-more-subsets into develop
2 files
+ 6
3
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -120,7 +120,7 @@ class AbstractPlotClass:
@@ -120,7 +120,7 @@ class AbstractPlotClass:
"""
"""
Standard colors used for train-, val-, and test-sets during postprocessing
Standard colors used for train-, val-, and test-sets during postprocessing
"""
"""
colors = {"train": "#e69f00", "val": "#009e73", "test": "#56b4e9"} # hex code
colors = {"train": "#e69f00", "val": "#009e73", "test": "#56b4e9", "train_val": "#000000"} # hex code
return colors
return colors
@@ -242,7 +242,7 @@ class PlotStationMap(AbstractPlotClass):
@@ -242,7 +242,7 @@ class PlotStationMap(AbstractPlotClass):
:width: 400
:width: 400
"""
"""
def __init__(self, generators: List, plot_folder: str = "."):
def __init__(self, generators: List, plot_folder: str = ".", plot_name="station_map"):
"""
"""
Set attributes and create plot.
Set attributes and create plot.
@@ -250,7 +250,7 @@ class PlotStationMap(AbstractPlotClass):
@@ -250,7 +250,7 @@ class PlotStationMap(AbstractPlotClass):
as value.
as value.
:param plot_folder: path to save the plot (default: current directory)
:param plot_folder: path to save the plot (default: current directory)
"""
"""
super().__init__(plot_folder, "station_map")
super().__init__(plot_folder, plot_name)
self._ax = None
self._ax = None
self._gl = None
self._gl = None
self._plot(generators)
self._plot(generators)
Loading