<commit_before>try:
  import torch
  import pytorch_lightning as pl  # noqa
  PYTORCH_LIGHTNING_IMPORT_FAILED = False
except ImportError:
  PYTORCH_LIGHTNING_IMPORT_FAILED = True


class DCLightningModule(pl.LightningModule):
  """DeepChem Lightning Module to be used with Lightning trainer.

  TODO: Add dataloader, example code and fit, once datasetmodule
  is ready<commit_msg>correctly import lightning to avoid import errors<commit_after>import torch
import pytorch_lightning as pl  # noqa


class DCLightningModule(pl.LightningModule):
  """DeepChem Lightning Module to be used with Lightning trainer.

  TODO: Add dataloader, example code and fit, once datasetmodule
  is ready
{'fraction_non_alphanumeric': 0.049707602339181284, 'fraction_numerical': 0.0, 'mean_word_length': 5.462264150943396, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '1453435', 'n_tokens_mistral': 214, 'n_tokens_neox': 207, 'n_words': 70}