Help with updating old code to new api

I’m trying to run a Jupyter notebook from this repository: https://github.com/artemyk/ibsgd. Specifically the MNIST_SaveActivations.ipynb file. Problem is the code is from two years ago, and The file was written in tensorflow 2.1.0. The notebook runs fine until the last cell, where I get the following error (see pictures below).

Error

loggingreporter.py file where the error occurs

​

How inputs variable is used in loggingreporter.py

I’ve been looking online but I can’t figure out how to fix this attribute error for the version of TensorFlow I’m running (2.8.0). Any guidance is appreciated! I’ve already tried downgrading to 2.1.0 but have not had success. I’m using the latest version of python3