code: def testInfoMax3DModular():
      import torch
      from deepchem.models.torch_models.gnn3d import InfoMax3DModular
  
      data, _ = get_regression_dataset()
  
      model = InfoMax3DModular(
          hidden_dim=64,
          target_dim=10,
          aggregators=["sum", "mean", "max"],
          readout_aggregators=["sum", "mean"],
          scalers=["identity"],
          device=torch.device("cpu"),
          task="pretraining",
          learning_rate=0.00001,
      )
  
      loss1 = model.fit(data, nb_epoch=1)
      loss2 = model.fit(data, nb_epoch=9)
      assert loss1 > loss2
  
file_path: deepchem/deepchem/models/tests/test_gnn3d.py
imports: import os
  import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: testInfoMax3DModular() without return types