          filename = f"{model_name}_state.{self._serializer.file_extension}"
          self._filename = os.path.join(model_dir, filename)
          self._name = model_name
          self._iterations = 0
          self._mixed_precision_layers: T.List[str] = []
          self._rebuild_model = False
          self._sessions: T.Dict[int, dict] = {}
          self._lowest_avg_loss: T.Dict[str, float] = {}
          self._config: T.Dict[str, ConfigValueType] = {}
          self._load(config_changeable_items)
 
</s> remove         self._sessions: T.Dict[int, dict] = {}
        self._lowest_avg_loss: T.Dict[str, float] = {}
        self._config: T.Dict[str, ConfigValueType] = {}
 </s> add         self._sessions: dict[int, dict] = {}
        self._lowest_avg_loss: dict[str, float] = {}
        self._config: dict[str, ConfigValueType] = {} </s> remove         self._inputs: T.List[tf.keras.layers.Layer] = []
        self._names: T.List[str] = []
        self._funcs: T.Dict[str, T.Callable] = {}
 </s> add         self._inputs: list[tf.keras.layers.Layer] = []
        self._names: list[str] = []
        self._funcs: dict[str, Callable] = {} </s> remove         self._stats: Dict[str, np.ndarray] = {}
 </s> add         self._stats: dict[str, np.ndarray] = {} </s> remove _CONFIG: T.Dict[str, ConfigValueType] = {}
 </s> add _CONFIG: dict[str, ConfigValueType] = {} </s> remove         self._args: Dict[str, Union[int, float]] = {"avg_samples": avg_samples,
                                                    "smooth_amount": smooth_amount,
                                                    "flatten_outliers": flatten_outliers}
 </s> add         self._args: dict[str, int | float] = {"avg_samples": avg_samples,
                                              "smooth_amount": smooth_amount,
                                              "flatten_outliers": flatten_outliers}
        self._mixed_precision_layers: list[str] = []
          filename = f"{model_name}_state.{self._serializer.file_extension}"
          self._filename = os.path.join(model_dir, filename)
          self._name = model_name
          self._iterations = 0
        self._mixed_precision_layers: list[str] = []
          self._rebuild_model = False
          self._sessions: T.Dict[int, dict] = {}
          self._lowest_avg_loss: T.Dict[str, float] = {}
          self._config: T.Dict[str, ConfigValueType] = {}
          self._load(config_changeable_items)
keep keep keep keep replace keep keep keep keep keep
 Rebase code (#1326)

* Remove tensorflow_probability requirement

* setup.py - fix progress bars

* requirements.txt: Remove pre python 3.9 packages

* update apple requirements.txt

* update INSTALL.md

* Remove python<3.9 code

* setup.py - fix Windows Installer

* typing: python3.9 compliant

* Update pytest and readthedocs python versions

* typing fixes

* Python Version updates
  - Reduce max version to 3.10
  - Default to 3.10 in installers
  - Remove incompatible 3.11 tests

* Update dependencies

* Downgrade imageio dep for Windows

* typing: merge optional unions and fixes

* Updates
  - min python version 3.10
  - typing to python 3.10 spec
  - remove pre-tf2.10 code
  - Add conda tests

* train: re-enable optimizer saving

* Update dockerfiles

* Update setup.py
  - Apple Conda deps to setup.py
  - Better Cuda + dependency handling

* bugfix: Patch logging to prevent Autograph errors

* Update dockerfiles

* Setup.py - Setup.py - stdout to utf-8

* Add more OSes to github Actions

* suppress mac-os end to end test 
plugins/train/model/_base/model.py