              self.display_item = Session
              self._add_trace_variables()
          else:
              logger.trace("Clearing graph")
              self.display_item = None
              self._clear_trace_variables()
 
</s> remove         loss_keys = [key
                     for key in self.display_item.get_loss_keys(Session.session_ids[-1])
                     if key != "total"]
 </s> add         loss_keys = self.display_item.get_loss_keys(Session.session_ids[-1])
        if not loss_keys:
            # Reload if we attempt to get loss keys before data is written
            logger.debug("Waiting for Session Data to become available to graph")
            self.after(1000, self.display_item_process)
            return

        loss_keys = [key for key in loss_keys if key != "total"] </s> remove         self._state = dict()
 </s> add         self._state = {} </s> remove         self._previewtrain = dict()
 </s> add         self._previewtrain = {} </s> remove             self._trace_vars = dict()
 </s> add             self._trace_vars = {} </s> remove         retval = dict()
 </s> add         retval = {}
        elif Session.is_training and self.display_item is not None:
            logger.trace("Graph already displayed. Nothing to do.")
              self.display_item = Session
              self._add_trace_variables()
        elif Session.is_training and self.display_item is not None:
            logger.trace("Graph already displayed. Nothing to do.")
          else:
              logger.trace("Clearing graph")
              self.display_item = None
              self._clear_trace_variables()
keep add keep keep keep keep
 Update code to support Tensorflow versions up to 2.8 (#1213)

* Update maximum tf version in setup + requirements

* - bump max version of tf version in launcher
- standardise tf version check

* update keras get_custom_objects  for tf>2.6

* bugfix: force black text in GUI file dialogs (linux)

* dssim loss - Move to stock tf.ssim function

* Update optimizer imports for compatibility

* fix logging for tf2.8

* Fix GUI graphing for TF2.8

* update tests

* bump requirements.txt versions

* Remove limit on nvidia-ml-py

* Graphing bugfixes
  - Prevent live graph from displaying if data not yet available

* bugfix: Live graph. Collect loss labels correctly

* fix: live graph - swallow inconsistent loss errors

* Bugfix: Prevent live graph from clearing during training

* Fix graphing for AMD 
lib/gui/display_command.py