class_docstr:     Pytorch Lightning module for semi-supervised learning.
  
  Args:
      active_dataset (ActiveLearningDataset): active learning dataset
      hparams (Namespace): hyper-parameters for the module
      **kwargs (**dict): extra arguments
class_docstr_tok: ["Pytorch","Lightning","module","for","semi-supervised","learning",".","Args",":","active_dataset","(","ActiveLearningDataset",")",":","active","learning","dataset","hparams","(","Namespace",")",":","hyper-parameters","for","the","module","*","*","kwargs","(","*","*","dict",")",":","extra","arguments"]
code: def supervised_training_step(self, batch, *args) -> Dict:
      raise NotImplementedError
  
file_path: baal/baal/utils/ssl_module.py
imports: import argparse
  import typing
  from argparse import Namespace
  from typing import Dict
  import pytorch_lightning
  from baal.active import ActiveLearningDataset
  from baal.utils.ssl_iterator import SemiSupervisedIterator
label_desc: Use image node_id 2 for calling the SSLModule obj's underlying member method code with example usage: obj.supervised_training_step(batch) without return types