Appropriate API integration

DimiChatzipavlis · DimiChatzipavlis · commit 5af4a44d12b0 · 2025-05-24T21:11:32.000+03:00
diff --git a/keras/src/callbacks/memory_usage_callback.py b/keras/src/callbacks/memory_usage_callback.py
@@ -4,6 +4,7 @@
 from keras.src.callbacks.callback import Callback
 from keras.src import backend as K
 
+# Attempt to import psutil for memory monitoring
 try:
     import psutil
 except ImportError:
