code: def apply_sharding(self, num_shards: int, shard_idx: int) -> None:
      self.range = range(shard_idx, len(self), num_shards)
  
imports: import copy
  from typing import Any, Callable, Iterator, Optional, Sequence
  import torch
  from torch_geometric.data import Batch
  from torch_geometric.utils import from_smiles
label_desc: Use image node_id 3 for calling the DatasetAdapter obj's underlying member method code with example usage: obj.apply_sharding(num_shards, shard_idx) without return types