addressing review comments

CHANGES.rst

@@ -21,7 +21,8 @@ New Features
 
 - The standalone version of jdaviz now uses solara instead of voila, resulting in faster load times. [#2909]
 
-- New configuration for ramp/Level 1 data products from Roman WFI and JWST [#3120, #3148, #3167, #3171, #3194]
+- New configuration for ramp/Level 1 and rate image/Level 2 data products from Roman WFI and
+  JWST [#3120, #3148, #3167, #3171, #3194]
 
 - Unit columns are now visible by default in the results table in model fitting. [#3196]
 

jdaviz/configs/rampviz/helper.py

@@ -114,11 +114,11 @@ def create_image_viewer(self, viewer_name=None, data=None):
             Image viewer instance.
 
         """
-        from jdaviz.configs.imviz.plugins.viewers import ImvizImageView
+        from jdaviz.configs.rampviz.plugins.viewers import RampvizImageView
 
         # Cannot assign data to real Data because it loads but it will
         # not update checkbox in Data menu.
 
         return self.app._on_new_viewer(
-            NewViewerMessage(ImvizImageView, data=None, sender=self.app),
+            NewViewerMessage(RampvizImageView, data=None, sender=self.app),
             vid=viewer_name, name=viewer_name)

jdaviz/configs/rampviz/plugins/viewers.py

@@ -131,7 +131,10 @@ def _initial_x_axis(self, *args):
         ref_data = self.state.reference_data
 
         if ref_data:
-            self.state.x_att = ref_data.id["Pixel Axis 0 [z]"]
+            if "Pixel Axis 0 [z]" in [comp.label for comp in ref_data.components]:
+                self.state.x_att = ref_data.id["Pixel Axis 0 [z]"]
+            else:
+                self.state.x_att = ref_data.id["Pixel Axis 0 [y]"]
 
     def set_plot_axes(self):
         self.figure.axes[1].tick_format = None