code: def load_param(self, params):
      key = f"{self.att_type}_{self.agg_type}_{self.in_channels}_{self.out_channels}_{self.heads}"
      weight_key = key + "_weight"
      att_key = key + "_att"
      agg_key = key + "_agg"
      bais_key = key + "_bais"
  
      if weight_key in params:
          self.weight = params[weight_key]
  
      if att_key in params:
          self.att = params[att_key]
  
      if bais_key in params:
          self.bias = params[bais_key]
  
      if agg_key in params and hasattr(self, "pool_layer"):
          self.pool_layer.load_state_dict(params[agg_key])
  
imports: import torch
  from torch import nn
  import torch.nn.functional
  import dgl
  from dgl.nn.pytorch.conv import *
  import dgl.function
  import math
  from dgl.utils import expand_as_pair
  from dgl.nn.functional import edge_softmax
  from torch.nn import Parameter
  import torch
  from torch import Tensor
  from ..scatter_utils import *
  from typing import Optional, Tuple, Union
label_desc: Use image node_id 10 for calling the GeoLayer obj's underlying member method code with example usage: obj.load_param(params) without return types