CDAT Migration: Update mp_partition_driver.py to use Dataset from dataset_xr.py

[tomvothecoder]

Description

• Related to [Refactor]: CDAT Migration Phase 3 - Port changes from main to cdat-migration-fy24 #871

Analyze files for CDAT migration

• mp_partition_driver.py -- Two lines for Dataset class from dataset.py, replace with Dataset class from dataset_xr.py
• mp_partition_parameter.py -- None
• mp_partition_parser.py -- None
• mp_partition_plot.py -- None
• mp_partition_viewer.py -- None

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• My changes generate no new warnings
• Any dependent changes have been merged and published in downstream modules

If applicable:

• New and existing unit tests pass with my changes (locally and CI/CD build)
• I have added tests that prove my fix is effective or that my feature works
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• I have noted that this is a breaking change for a major release (fix or feature that would cause existing functionality to not work as expected)

[tomvothecoder]

Hi @chengzhuzhang, I analyzed all of the mp_partition related modules and only found these changes. Does this set run and if yes, should I test it? I did not try yet.

[chengzhuzhang]

Hi @chengzhuzhang, I analyzed all of the mp_partition related modules and only found these changes. Does this set run and if yes, should I test it? I did not try yet.

This set should run as is, just there is a new enhancement planned to add a new figure that uses new variables from v3 output. Maybe have it a try to see if the code refactor works. Note that this set requires monthly time-series of a set of variables such as, LANDFRAC, T, CLDICE, CLDLIQ exist in test_data_path.

[tomvothecoder]

Hi @chengzhuzhang, I analyzed all of the mp_partition related modules and only found these changes. Does this set run and if yes, should I test it? I did not try yet.

This set should run as is, just there is a new enhancement planned to add a new figure that uses new variables from v3 output. Maybe have it a try to see if the code refactor works. Note that this set requires monthly time-series of a set of variables such as, LANDFRAC, T, CLDICE, CLDLIQ exist in test_data_path.

I forgot that "mp_partition" has been running by default in the run_all..py script. I just ran the code and it successfully generates the plots. There are no visible differences as far as I can tell.

• cdat-migration-fy24: https://portal.nersc.gov/project/e3sm/cdat-migration-fy24/871-mp-partition/mp_partition/mixed-phase_partition/mixed-phase_partition.png
• main: https://portal.nersc.gov/project/e3sm/cdat-migration-fy24/main/mp_partition/mixed-phase_partition/mixed-phase_partition.png
• Diff: https://portal.nersc.gov/project/e3sm/cdat-migration-fy24/871-mp-partition/mp_partition/mixed-phase_partition_diff/mixed-phase_partition.png

I will merge this PR once the build passes.