From c909cfd3d1c0885c97dddadc076108cde2513131 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Fri, 3 Mar 2023 22:45:34 +0000
Subject: [PATCH] fixup

---
 aimodel/src/lib/ai/components/MetricMeanIoU.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/aimodel/src/lib/ai/components/MetricMeanIoU.py b/aimodel/src/lib/ai/components/MetricMeanIoU.py
index 470220f..6469472 100644
--- a/aimodel/src/lib/ai/components/MetricMeanIoU.py
+++ b/aimodel/src/lib/ai/components/MetricMeanIoU.py
@@ -3,9 +3,9 @@ import math
 import tensorflow as tf
 
 
-def make_one_hot_mean_iou():
+def make_one_hot_mean_iou(classes=2):
 	iou = tf.keras.metrics.MeanIoU(num_classes=classes)
-	def one_hot_mean_iou(y_true, y_pred, classes=2):
+	def one_hot_mean_iou(y_true, y_pred, ):
 		"""Compute the mean IoU for one-hot tensors.
 		Args:
 			y_true (tf.Tensor): The ground truth label.