class_docstr: Class used to draw a circle.
class_docstr_tok: ["Class","used","to","draw","a","circle","."]
docstr: Set axis.
docstr_tok: ["Set","axis","."]
code: def set_ax(ax=None):
      """Set axis."""
      if ax is None:
          ax = plt.subplot()
      ax_s = AX_SCALE
      plt.setp(
          ax,
          xlim=(-ax_s, ax_s),
          ylim=(-ax_s, ax_s),
          xlabel="X",
          ylabel="Y",
      )
      return ax
  
file_path: geomstats/geomstats/visualization/hypersphere.py
imports: import matplotlib.pyplot
  from mpl_toolkits.mplot3d import Axes3D
  import geomstats.backend
  from geomstats.geometry.hypersphere import Hypersphere
label_desc: Use image node_id 2 for calling the Circle obj's underlying member method code with example usage: obj.set_ax(ax) and returns: ax