From 9074fd11de2016cab8f75889231a8f448e29a868 Mon Sep 17 00:00:00 2001
From: dkupnicki <97047305+dkupnicki@users.noreply.github.com>
Date: Wed, 14 Aug 2024 15:07:52 +0200
Subject: [PATCH] add **kwargs

---
 .../semantic_segmentation/unet_3d/brats_19/run.py           | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/computer_vision/semantic_segmentation/unet_3d/brats_19/run.py b/computer_vision/semantic_segmentation/unet_3d/brats_19/run.py
index 08300141..e09a5f5e 100644
--- a/computer_vision/semantic_segmentation/unet_3d/brats_19/run.py
+++ b/computer_vision/semantic_segmentation/unet_3d/brats_19/run.py
@@ -61,15 +61,15 @@ def run_single_pass(tf_runner, brats):
     return run_model(run_single_pass, runner, dataset, 1, num_runs, timeout)
 
 
-def run_tf_fp32(model_path, num_runs, timeout, dataset_path):
+def run_tf_fp32(model_path, num_runs, timeout, dataset_path, **kwargs):
     return run_tf_fp(model_path, num_runs, timeout, dataset_path)
 
 
-def run_tf_fp16(model_path, num_runs, timeout, dataset_path):
+def run_tf_fp16(model_path, num_runs, timeout, dataset_path, **kwargs):
     return run_tf_fp(model_path, num_runs, timeout, dataset_path)
 
 
-def run_pytorch_fp32(model_path, num_runs, timeout, dataset_path):
+def run_pytorch_fp32(model_path, num_runs, timeout, dataset_path, **kwargs):
     import torch
     import numpy as np
     from utils.cv.brats import BraTS19