From 726dbcc7cae0abc47cc3a9846c77f0d6c4f20c3e Mon Sep 17 00:00:00 2001
From: Jameson Graef Rollins <jrollins@finestructure.net>
Date: Thu, 13 Feb 2020 08:29:37 -0800
Subject: [PATCH] ifo: add missing plot_style from canonical budgets

---
 gwinc/ifo/Aplus/__init__.py   | 3 +++
 gwinc/ifo/CE1/__init__.py     | 7 +++++++
 gwinc/ifo/CE2/__init__.py     | 7 +++++++
 gwinc/ifo/Voyager/__init__.py | 3 +++
 gwinc/ifo/__init__.py         | 1 +
 gwinc/ifo/aLIGO/__init__.py   | 3 +++
 6 files changed, 24 insertions(+)

diff --git a/gwinc/ifo/Aplus/__init__.py b/gwinc/ifo/Aplus/__init__.py
index 93375518..78cf3c30 100644
--- a/gwinc/ifo/Aplus/__init__.py
+++ b/gwinc/ifo/Aplus/__init__.py
@@ -1,4 +1,5 @@
 from gwinc.ifo.noises import *
+from gwinc.ifo import PLOT_STYLE
 
 
 class Aplus(nb.Budget):
@@ -20,3 +21,5 @@ class Aplus(nb.Budget):
     calibrations = [
         Strain,
     ]
+
+    plot_style = PLOT_STYLE
diff --git a/gwinc/ifo/CE1/__init__.py b/gwinc/ifo/CE1/__init__.py
index d9ff6474..442262be 100644
--- a/gwinc/ifo/CE1/__init__.py
+++ b/gwinc/ifo/CE1/__init__.py
@@ -1,4 +1,6 @@
 from gwinc.ifo.noises import *
+from gwinc.ifo import PLOT_STYLE
+
 
 class Newtonian(nb.Budget):
     """Newtonian Gravity
@@ -18,6 +20,7 @@ class Newtonian(nb.Budget):
         NewtonianInfrasound,
     ]
 
+
 class Coating(nb.Budget):
     """Coating Thermal
 
@@ -35,6 +38,7 @@ class Coating(nb.Budget):
         CoatingThermoOptic,
     ]
 
+
 class Substrate(nb.Budget):
     """Substrate Thermal
 
@@ -53,6 +57,7 @@ class Substrate(nb.Budget):
         SubstrateThermoElastic,
     ]
 
+
 class CE1(nb.Budget):
 
     name = 'Cosmic Explorer 1'
@@ -70,3 +75,5 @@ class CE1(nb.Budget):
     calibrations = [
         Strain,
     ]
+
+    plot_style = PLOT_STYLE
diff --git a/gwinc/ifo/CE2/__init__.py b/gwinc/ifo/CE2/__init__.py
index 7d63e2f5..34fc40c1 100644
--- a/gwinc/ifo/CE2/__init__.py
+++ b/gwinc/ifo/CE2/__init__.py
@@ -1,4 +1,6 @@
 from gwinc.ifo.noises import *
+from gwinc.ifo import PLOT_STYLE
+
 
 class Newtonian(nb.Budget):
     """Newtonian Gravity
@@ -18,6 +20,7 @@ class Newtonian(nb.Budget):
         NewtonianInfrasound,
     ]
 
+
 class Coating(nb.Budget):
     """Coating Thermal
 
@@ -35,6 +38,7 @@ class Coating(nb.Budget):
         CoatingThermoOptic,
     ]
 
+
 class Substrate(nb.Budget):
     """Substrate Thermal
 
@@ -55,6 +59,7 @@ class Substrate(nb.Budget):
         SubstrateThermoElastic,
     ]
 
+
 class CE2(nb.Budget):
 
     name = 'Cosmic Explorer 2'
@@ -72,3 +77,5 @@ class CE2(nb.Budget):
     calibrations = [
         Strain,
     ]
+
+    plot_style = PLOT_STYLE
diff --git a/gwinc/ifo/Voyager/__init__.py b/gwinc/ifo/Voyager/__init__.py
index f577b624..933a7828 100644
--- a/gwinc/ifo/Voyager/__init__.py
+++ b/gwinc/ifo/Voyager/__init__.py
@@ -1,4 +1,5 @@
 from gwinc.ifo.noises import *
+from gwinc.ifo import PLOT_STYLE
 
 
 class Voyager(nb.Budget):
@@ -22,3 +23,5 @@ class Voyager(nb.Budget):
     calibrations = [
         Strain,
     ]
+
+    plot_style = PLOT_STYLE
diff --git a/gwinc/ifo/__init__.py b/gwinc/ifo/__init__.py
index 281a5a05..e42c608d 100644
--- a/gwinc/ifo/__init__.py
+++ b/gwinc/ifo/__init__.py
@@ -9,6 +9,7 @@ IFOS = [
     'CE2',
 ]
 
+
 PLOT_STYLE = dict(
     ylabel=u"Strain [1/\u221AHz]",
 )
diff --git a/gwinc/ifo/aLIGO/__init__.py b/gwinc/ifo/aLIGO/__init__.py
index 070b0c07..dcbdaa8a 100644
--- a/gwinc/ifo/aLIGO/__init__.py
+++ b/gwinc/ifo/aLIGO/__init__.py
@@ -1,4 +1,5 @@
 from gwinc.ifo.noises import *
+from gwinc.ifo import PLOT_STYLE
 
 
 class aLIGO(nb.Budget):
@@ -20,3 +21,5 @@ class aLIGO(nb.Budget):
     calibrations = [
         Strain,
     ]
+
+    plot_style = PLOT_STYLE
-- 
GitLab