code: def test_deepstate_smoke(estimator, datasets):
      # TODO: pass `dtype` below once you add `dtype` support to `DeepStateEstimator`
      estimator = estimator()
      dataset_train, dataset_test = datasets
      predictor = estimator.train(dataset_train)
      forecasts = list(predictor.predict(dataset_test))
      assert len(forecasts) == len(dataset_test)
  
imports: from functools import partial
  import pytest
  from gluonts.mx.model.deepstate import DeepStateEstimator
  from gluonts.mx.trainer import Trainer
  from gluonts.testutil.dummy_datasets import make_dummy_datasets_with_features
label_desc: Use image node_id 1 for calling a global function with example usage: test_deepstate_smoke(estimator, datasets) without return types