uh. still need netcdf for xarray to read netcdf files.

pyproject.toml

@@ -32,6 +32,7 @@ dependencies = [
  "NREL-farms>=1.0.4",
  "dask>=2022.0",
  "google-auth-oauthlib==0.5.3",
+ "h5netcdf",
  "matplotlib>=3.1",
  "numpy>=1.7.0",
  "pandas>=2.0",

sup3r/models/base.py

@@ -688,7 +688,8 @@ def train_epoch(self,
 
  if only_gen or (train_gen and not gen_too_good):
  trained_gen = True
- b_loss_details = self.run_gradient_descent(
+ b_loss_details = self.timer(
+ self.run_gradient_descent,
  batch.low_res,
  batch.high_res,
  self.generator_weights,
@@ -700,7 +701,8 @@ def train_epoch(self,
 
  if only_disc or (train_disc and not disc_too_good):
  trained_disc = True
- b_loss_details = self.run_gradient_descent(
+ b_loss_details = self.timer(
+ self.run_gradient_descent,
  batch.low_res,
  batch.high_res,
  self.discriminator_weights,

sup3r/utilities/__init__.py

@@ -3,6 +3,7 @@
 from enum import Enum
 
 import dask
+import netCDF4
 import numpy as np
 import pandas as pd
 import phygnn
@@ -22,6 +23,7 @@
  'nrel-rex': rex.__version__,
  'python': sys.version,
  'xarray': xarray.__version__,
+ 'netCDF4': netCDF4.__version__,
  'dask': dask.__version__,
  }