code: def testInfoMax3DModularRegression():
      import torch
      from deepchem.models.torch_models.gnn3d import InfoMax3DModular
  
      data, metric = get_regression_dataset()
  
      model = InfoMax3DModular(
          hidden_dim=64,
          aggregators=["sum", "mean", "max"],
          readout_aggregators=["sum", "mean"],
          scalers=["identity"],
          task="regression",
          n_tasks=1,
          device=torch.device("cpu"),
      )
  
      model.fit(data, nb_epoch=100)
      scores = model.evaluate(data, [metric])
      print(scores)
      assert scores["mean_absolute_error"] < 0.5
  
file_path: deepchem/deepchem/models/tests/test_gnn3d.py
imports: import os
  import pytest
label_desc: Use image node_id 8 for calling a global function with example usage: testInfoMax3DModularRegression() without return types