code: def _get_axis_dims(self, element):
      if isinstance(element, Graph):
          element = element.nodes
      return element.dimensions()[:2]
  
imports: import numpy
  from ..core import Dataset, Dimension, util
  from ..element import Bars, Graph
  from ..element.util import categorical_aggregate2d
  from .util import get_axis_padding
label_desc: Use image node_id 1 for calling the GraphMixin obj's underlying member method code with example usage: obj._get_axis_dims(element) and returns: element