code: def test_infographstar_fit_restore():
      from deepchem.models.torch_models.infograph import (
          InfoGraphStarModel,
      )
  
      dataset, _ = get_classification_dataset()
      num_feat = 30
      edge_dim = 11
      dim = 64
  
      model = InfoGraphStarModel(
          num_feat, edge_dim, dim, task="supervised"
      )
  
      model.fit(dataset, nb_epoch=100)
  
      model2 = InfoGraphStarModel(
          num_feat,
          edge_dim,
          dim,
          training_mode="supervised",
          model_dir=model.model_dir,
      )
      model2.fit(dataset, nb_epoch=1, restore=True)
      prediction = model2.predict_on_batch(dataset.X).reshape(-1, 1)
      assert np.allclose(dataset.y, np.round(prediction))
  
imports: import os
  import pytest
  import numpy
  import deepchem
  from deepchem.feat.molecule_featurizers import MolGraphConvFeaturizer
label_desc: Use image node_id 15 for calling a global function with example usage: test_infographstar_fit_restore() without return types