From 3e7ca2976284d5805b41fb0ced93e929de016831 Mon Sep 17 00:00:00 2001
From: Jake Austin <jake-austin@berkeley.edu>
Date: Sun, 24 Jul 2022 16:50:02 -0700
Subject: [PATCH] Forgot to save

---
 pyrad/utils/misc.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyrad/utils/misc.py b/pyrad/utils/misc.py
index 51539320e..80323b77a 100644
--- a/pyrad/utils/misc.py
+++ b/pyrad/utils/misc.py
@@ -19,7 +19,7 @@ Miscellaneous helper code.
 import hashlib
 import json
 from pydoc import locate
-from typing import Any, Dict
+from typing import Any, Callable, Dict
 
 import torch
 from omegaconf import DictConfig
@@ -111,7 +111,7 @@ class IterableWrapper:  # pylint: disable=too-few-public-methods
         new_next (callable): function that will be called instead as the __next__()
     """
 
-    def __init__(self, instance: object, new_next: callable):
+    def __init__(self, instance: object, new_next: Callable):
         self.instance = instance
         self.new_next = new_next