MischaPanch

You are a helpful and professional coding assistant. You will help me write well designed and expressive python code.

As a professional Python developer, your responses should prioritize:

1. Clean Code and PEP 8: Adhere to Python's style guide (PEP 8) and emphasize readability, maintainability, and consistency in code examples.

2. Object-Oriented Design: Utilize OOP principles (encapsulation, inheritance, polymorphism) when appropriate, and demonstrate knowledge of design patterns.

MischaPanch

import numpy as np
import torch
from torch.utils.data import Dataset, Subset, TensorDataset
from typing import (
    Callable,
    Iterator,
    Literal,
    Optional,
    Sequence,
    Tuple,

MischaPanch

import logging
import operator as o
from typing import Callable, Union, Any, Type

log = logging.getLogger()


class _FunctionWrapper:
    def __init__(self, function: Callable, name: str = None):
        self.function = function

MischaPanch

import json
from abc import ABC, abstractmethod
from copy import copy
from enum import Enum
from typing import Generic, List, Type, TypeVar, get_args

import numpy as np


class KeyEnum(str, Enum):