code: def wrap_oss_with_dataclass(union):
      if not IS_FB_ENVIRONMENT:
          # OSS TaggedUnion
          return dataclass(frozen=True)(union)
      else:
          # FBL TaggedUnion
          return union
  
file_path: ReAgent/reagent/core/registry_meta.py
imports: import abc
  import logging
  import os
  from typing import Dict, Optional, Type
  from reagent.core.dataclasses import dataclass
  from reagent.core.fb_checker import IS_FB_ENVIRONMENT
label_desc: Use image node_id 2 for calling a global function with example usage: wrap_oss_with_dataclass(union) and returns: dataclass, union