Skip to content

Commit 72bf515

Browse files
Fix import bug with model_eval=physical
1 parent 417d8cf commit 72bf515

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

wf_psf/script_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,8 @@ def evaluate_model(**args):
549549
test_SEDs = test_dataset['SEDs']
550550
# test_zernike_coef = test_dataset['zernike_coef']
551551
# ground truth d_max (spatial polynomial max order)
552-
d_max_gt = test_dataset['parameters']['d_max']
552+
if args['model_eval'] == 'poly':
553+
d_max_gt = test_dataset['parameters']['d_max']
553554

554555
# Convert to tensor
555556
tf_noisy_train_stars = tf.convert_to_tensor(train_dataset['noisy_stars'], dtype=tf.float32)

0 commit comments

Comments
 (0)