From a34ca6a94a1c5b852e21828fbd2371693229dde4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Math=C3=AFs=20F=C3=A9d=C3=A9rico?= Date: Fri, 5 Jan 2024 16:36:43 +0100 Subject: [PATCH] =?UTF-8?q?=20=F0=9F=93=9D=20Update=20environment=20exampl?= =?UTF-8?q?es=20list=20=F0=9F=8E=89=20Add=20cli=20for=20minicraft=20&=20tr?= =?UTF-8?q?easure=20envs=20=F0=9F=8E=89=20Add=20gym=20register=20for=20tre?= =?UTF-8?q?asure=20env=20=F0=9F=94=A7=20Fix=20ruff=20config?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .pylintrc | 510 --------------------- pyproject.toml | 4 +- src/hcraft/cli.py | 44 ++ src/hcraft/examples/__init__.py | 30 +- src/hcraft/examples/minicraft/__init__.py | 6 +- src/hcraft/examples/minicraft/minicraft.py | 6 +- src/hcraft/examples/treasure/__init__.py | 13 + 7 files changed, 91 insertions(+), 522 deletions(-) delete mode 100644 .pylintrc diff --git a/.pylintrc b/.pylintrc deleted file mode 100644 index dd16b764..00000000 --- a/.pylintrc +++ /dev/null @@ -1,510 +0,0 @@ -[MASTER] - -# A comma-separated list of package or module names from where C extensions may -# be loaded. Extensions are loading into the active Python interpreter and may -# run arbitrary code. -extension-pkg-whitelist= - -# Specify a score threshold to be exceeded before program exits with error. -fail-under=10.0 - -# Add files or directories to the blacklist. They should be base names, not -# paths. -ignore=CVS - -# Add files or directories matching the regex patterns to the blacklist. The -# regex matches against base names, not paths. -ignore-patterns= - -# Python code to execute, usually for sys.path manipulation such as -# pygtk.require(). -#init-hook= - -# Use multiple processes to speed up Pylint. Specifying 0 will auto-detect the -# number of processors available to use. -jobs=1 - -# Control the amount of potential inferred values when inferring a single -# object. This can help the performance when dealing with large functions or -# complex, nested conditions. -limit-inference-results=100 - -# List of plugins (as comma separated values of python module names) to load, -# usually to register additional checkers. -load-plugins= - -# Pickle collected data for later comparisons. -persistent=yes - -# When enabled, pylint would attempt to guess common misconfiguration and emit -# user-friendly hints instead of false-positive error messages. -suggestion-mode=yes - -# Allow loading of arbitrary C extensions. Extensions are imported into the -# active Python interpreter and may run arbitrary code. -unsafe-load-any-extension=no - - -[MESSAGES CONTROL] - -# Only show warnings with the listed confidence levels. Leave empty to show -# all. Valid levels: HIGH, INFERENCE, INFERENCE_FAILURE, UNDEFINED. -confidence= - -# Disable the message, report, category or checker with the given id(s). You -# can either give multiple identifiers separated by comma (,) or put this -# option multiple times (only on the command line, not in the configuration -# file where it should appear only once). You can also use "--disable=all" to -# disable everything first and then reenable specific checks. For example, if -# you want to run only the similarities checker, you can use "--disable=all -# --enable=similarities". If you want to run only the classes checker, but have -# no Warning level messages displayed, use "--disable=all --enable=classes -# --disable=W". -disable= - -# Enable the message, report, category or checker with the given id(s). You can -# either give multiple identifier separated by comma (,) or put this option -# multiple time (only on the command line, not in the configuration file where -# it should appear only once). See also the "--disable" option for examples. -enable=c-extension-no-member - - -[REPORTS] - -# Python expression which should return a score less than or equal to 10. You -# have access to the variables 'error', 'warning', 'refactor', and 'convention' -# which contain the number of messages in each category, as well as 'statement' -# which is the total number of statements analyzed. This score is used by the -# global evaluation report (RP0004). -evaluation=10.0 - ((float(5 * error + warning + refactor + convention) / statement) * 10) - -# Template used to display messages. This is a python new-style format string -# used to format the message information. See doc for all details. -#msg-template= - -# Set the output format. Available formats are text, parseable, colorized, json -# and msvs (visual studio). You can also give a reporter class, e.g. -# mypackage.mymodule.MyReporterClass. -output-format=text - -# Tells whether to display a full report or only the messages. -reports=no - -# Activate the evaluation score. -score=yes - - -[REFACTORING] - -# Maximum number of nested blocks for function / method body -max-nested-blocks=5 - -# Complete name of functions that never returns. When checking for -# inconsistent-return-statements if a never returning function is called then -# it will be considered as an explicit return statement and no message will be -# printed. -never-returning-functions=sys.exit - - -[BASIC] - -# Naming style matching correct argument names. -argument-naming-style=snake_case - -# Regular expression matching correct argument names. Overrides argument- -# naming-style. -#argument-rgx= - -# Naming style matching correct attribute names. -attr-naming-style=snake_case - -# Regular expression matching correct attribute names. Overrides attr-naming- -# style. -#attr-rgx= - -# Bad variable names which should always be refused, separated by a comma. -bad-names=foo, - bar, - baz, - toto, - tutu, - tata - -# Bad variable names regexes, separated by a comma. If names match any regex, -# they will always be refused -bad-names-rgxs= - -# Naming style matching correct class attribute names. -class-attribute-naming-style=any - -# Regular expression matching correct class attribute names. Overrides class- -# attribute-naming-style. -#class-attribute-rgx= - -# Naming style matching correct class names. -class-naming-style=PascalCase - -# Regular expression matching correct class names. Overrides class-naming- -# style. -#class-rgx= - -# Naming style matching correct constant names. -const-naming-style=UPPER_CASE - -# Regular expression matching correct constant names. Overrides const-naming- -# style. -#const-rgx= - -# Minimum line length for functions/classes that require docstrings, shorter -# ones are exempt. -docstring-min-length=-1 - -# Naming style matching correct function names. -function-naming-style=snake_case - -# Regular expression matching correct function names. Overrides function- -# naming-style. -#function-rgx= - -# Good variable names which should always be accepted, separated by a comma. -good-names=i, - j, - k, - ex, - Run, - _ - -# Good variable names regexes, separated by a comma. If names match any regex, -# they will always be accepted -good-names-rgxs= - -# Include a hint for the correct naming format with invalid-name. -include-naming-hint=no - -# Naming style matching correct inline iteration names. -inlinevar-naming-style=any - -# Regular expression matching correct inline iteration names. Overrides -# inlinevar-naming-style. -#inlinevar-rgx= - -# Naming style matching correct method names. -method-naming-style=snake_case - -# Regular expression matching correct method names. Overrides method-naming- -# style. -#method-rgx= - -# Naming style matching correct module names. -module-naming-style=snake_case - -# Regular expression matching correct module names. Overrides module-naming- -# style. -#module-rgx= - -# Colon-delimited sets of names that determine each other's naming style when -# the name regexes allow several styles. -name-group= - -# Regular expression which should only match function or class names that do -# not require a docstring. -no-docstring-rgx=^_ - -# List of decorators that produce properties, such as abc.abstractproperty. Add -# to this list to register other decorators that produce valid properties. -# These decorators are taken in consideration only for invalid-name. -property-classes=abc.abstractproperty - -# Naming style matching correct variable names. -variable-naming-style=snake_case - -# Regular expression matching correct variable names. Overrides variable- -# naming-style. -#variable-rgx= - - -[FORMAT] - -# Expected format of line ending, e.g. empty (any line ending), LF or CRLF. -expected-line-ending-format= - -# Regexp for a line that is allowed to be longer than the limit. -ignore-long-lines=^\s*(# )??$ - -# Number of spaces of indent required inside a hanging or continued line. -indent-after-paren=4 - -# String used as indentation unit. This is usually " " (4 spaces) or "\t" (1 -# tab). -indent-string=' ' - -# Maximum number of characters on a single line. -max-line-length=100 - -# Maximum number of lines in a module. -max-module-lines=1000 - -# Allow the body of a class to be on the same line as the declaration if body -# contains single statement. -single-line-class-stmt=no - -# Allow the body of an if to be on the same line as the test if there is no -# else. -single-line-if-stmt=no - - -[LOGGING] - -# The type of string formatting that logging methods do. `old` means using % -# formatting, `new` is for `{}` formatting. -logging-format-style=old - -# Logging modules to check that the string format arguments are in logging -# function parameter format. -logging-modules=logging - - -[MISCELLANEOUS] - -# List of note tags to take in consideration, separated by a comma. -notes=FIXME, - XXX, - TODO - -# Regular expression of note tags to take in consideration. -#notes-rgx= - - -[SIMILARITIES] - -# Ignore comments when computing similarities. -ignore-comments=yes - -# Ignore docstrings when computing similarities. -ignore-docstrings=yes - -# Ignore imports when computing similarities. -ignore-imports=no - -# Minimum lines number of a similarity. -min-similarity-lines=4 - - -[SPELLING] - -# Limits count of emitted suggestions for spelling mistakes. -max-spelling-suggestions=4 - -# Spelling dictionary name. Available dictionaries: none. To make it work, -# install the python-enchant package. -spelling-dict= - -# List of comma separated words that should not be checked. -spelling-ignore-words= - -# A path to a file that contains the private dictionary; one word per line. -spelling-private-dict-file= - -# Tells whether to store unknown words to the private dictionary (see the -# --spelling-private-dict-file option) instead of raising a message. -spelling-store-unknown-words=no - - -[STRING] - -# This flag controls whether inconsistent-quotes generates a warning when the -# character used as a quote delimiter is used inconsistently within a module. -check-quote-consistency=no - -# This flag controls whether the implicit-str-concat should generate a warning -# on implicit string concatenation in sequences defined over several lines. -check-str-concat-over-line-jumps=no - - -[TYPECHECK] - -# List of decorators that produce context managers, such as -# contextlib.contextmanager. Add to this list to register other decorators that -# produce valid context managers. -contextmanager-decorators=contextlib.contextmanager - -# List of members which are set dynamically and missed by pylint inference -# system, and so shouldn't trigger E1101 when accessed. Python regular -# expressions are accepted. -generated-members= - -# Tells whether missing members accessed in mixin class should be ignored. A -# mixin class is detected if its name ends with "mixin" (case insensitive). -ignore-mixin-members=yes - -# Tells whether to warn about missing members when the owner of the attribute -# is inferred to be None. -ignore-none=yes - -# This flag controls whether pylint should warn about no-member and similar -# checks whenever an opaque object is returned when inferring. The inference -# can return multiple potential results while evaluating a Python object, but -# some branches might not be evaluated, which results in partial inference. In -# that case, it might be useful to still emit no-member and other checks for -# the rest of the inferred objects. -ignore-on-opaque-inference=yes - -# List of class names for which member attributes should not be checked (useful -# for classes with dynamically set attributes). This supports the use of -# qualified names. -ignored-classes=optparse.Values,thread._local,_thread._local - -# List of module names for which member attributes should not be checked -# (useful for modules/projects where namespaces are manipulated during runtime -# and thus existing member attributes cannot be deduced by static analysis). It -# supports qualified module names, as well as Unix pattern matching. -ignored-modules= - -# Show a hint with possible names when a member name was not found. The aspect -# of finding the hint is based on edit distance. -missing-member-hint=yes - -# The minimum edit distance a name should have in order to be considered a -# similar match for a missing member name. -missing-member-hint-distance=1 - -# The total number of similar names that should be taken in consideration when -# showing a hint for a missing member. -missing-member-max-choices=1 - -# List of decorators that change the signature of a decorated function. -signature-mutators= - - -[VARIABLES] - -# List of additional names supposed to be defined in builtins. Remember that -# you should avoid defining new builtins when possible. -additional-builtins= - -# Tells whether unused global variables should be treated as a violation. -allow-global-unused-variables=yes - -# List of strings which can identify a callback function by name. A callback -# name must start or end with one of those strings. -callbacks=cb_, - _cb - -# A regular expression matching the name of dummy variables (i.e. expected to -# not be used). -dummy-variables-rgx=_+$|(_[a-zA-Z0-9_]*[a-zA-Z0-9]+?$)|dummy|^ignored_|^unused_ - -# Argument names that match this expression will be ignored. Default to name -# with leading underscore. -ignored-argument-names=_.*|^ignored_|^unused_ - -# Tells whether we should check for unused import in __init__ files. -init-import=no - -# List of qualified module names which can have objects that can redefine -# builtins. -redefining-builtins-modules=six.moves,past.builtins,future.builtins,builtins,io - - -[CLASSES] - -# List of method names used to declare (i.e. assign) instance attributes. -defining-attr-methods=__init__, - __new__, - setUp, - __post_init__ - -# List of member names, which should be excluded from the protected access -# warning. -exclude-protected=_asdict, - _fields, - _replace, - _source, - _make - -# List of valid names for the first argument in a class method. -valid-classmethod-first-arg=cls - -# List of valid names for the first argument in a metaclass class method. -valid-metaclass-classmethod-first-arg=cls - - -[DESIGN] - -# Maximum number of arguments for function / method. -max-args=10 - -# Maximum number of attributes for a class (see R0902). -max-attributes=20 - -# Maximum number of boolean expressions in an if statement (see R0916). -max-bool-expr=5 - -# Maximum number of branch for function / method body. -max-branches=20 - -# Maximum number of locals for function / method body. -max-locals=25 - -# Maximum number of parents for a class (see R0901). -max-parents=7 - -# Maximum number of public methods for a class (see R0904). -max-public-methods=20 - -# Maximum number of return / yield for function / method body. -max-returns=6 - -# Maximum number of statements in function / method body. -max-statements=50 - -# Minimum number of public methods for a class (see R0903). -min-public-methods=2 - - -[IMPORTS] - -# List of modules that can be imported at any level, not just the top level -# one. -allow-any-import-level= - -# Allow wildcard imports from modules that define __all__. -allow-wildcard-with-all=no - -# Analyse import fallback blocks. This can be used to support both Python 2 and -# 3 compatible code, which means that the block might have code that exists -# only in one or another interpreter, leading to false positives when analysed. -analyse-fallback-blocks=no - -# Deprecated modules which should not be used, separated by a comma. -deprecated-modules=optparse,tkinter.tix - -# Create a graph of external dependencies in the given file (report RP0402 must -# not be disabled). -ext-import-graph= - -# Create a graph of every (i.e. internal and external) dependencies in the -# given file (report RP0402 must not be disabled). -import-graph= - -# Create a graph of internal dependencies in the given file (report RP0402 must -# not be disabled). -int-import-graph= - -# Force import order to recognize a module as part of the standard -# compatibility libraries. -known-standard-library= - -# Force import order to recognize a module as part of a third party library. -known-third-party=enchant - -# Couples of modules and preferred modules, separated by a comma. -preferred-modules= - - -[EXCEPTIONS] - -# Exceptions that will emit a warning when being caught. Defaults to -# "BaseException, Exception". -overgeneral-exceptions=BaseException, - Exception diff --git a/pyproject.toml b/pyproject.toml index cb33b03f..fead54c2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -56,7 +56,7 @@ source = ["src"] [tool.ruff] # Enable pycodestyle (`E`) and Pyflakes (`F`) codes by default. select = ["E", "F"] -ignore = [] +ignore = ["E501"] # Allow autofix for all enabled rules (when `--fix`) is provided. fixable = ["A", "B", "C", "D", "E", "F"] @@ -87,7 +87,7 @@ exclude = [ ] # Same as Black. -line-length = 120 +line-length = 88 # Allow unused variables when underscore-prefixed. dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$" diff --git a/src/hcraft/cli.py b/src/hcraft/cli.py index f8c3c8e0..11263ce8 100644 --- a/src/hcraft/cli.py +++ b/src/hcraft/cli.py @@ -10,6 +10,8 @@ RecursiveHcraftEnv, TowerHcraftEnv, ) +from hcraft.examples.minicraft import MINICRAFT_NAME_TO_ENV +from hcraft.examples.treasure import TreasureEnv from hcraft.purpose import Purpose from hcraft.render.render import HcraftWindow from hcraft.render.widgets import ContentMode, DisplayMode @@ -26,9 +28,11 @@ def hcraft_cli(args: Optional[List[str]] = None) -> HcraftEnv: subparsers = parser.add_subparsers(help="Available environments") _minecraft_sub_parser(subparsers) + _minicraft_sub_parser(subparsers) _tower_sub_parser(subparsers) _recursive_sub_parser(subparsers) _light_recursive_sub_parser(subparsers) + _treasure_sub_parser(subparsers) _random_sub_parser(subparsers) parser.add_argument( @@ -148,6 +152,28 @@ def _minehcraft_from_cli(args: Namespace): return env +def _minicraft_sub_parser(subparsers: "_SubParsersAction[ArgumentParser]"): + subparser = subparsers.add_parser( + "minicraft", + help="MiniHcraft: Bunch of environments inspired from MiniGrid environments.", + ) + subparser.add_argument( + "name", + type=str, + help="Name of the minicraft environment.", + choices=MINICRAFT_NAME_TO_ENV.keys(), + ) + subparser.set_defaults(func=_minicraft_from_cli) + + +def _minicraft_from_cli(args: Namespace): + window = _window_from_cli(args) + return MINICRAFT_NAME_TO_ENV[args.name]( + render_window=window, + max_step=args.max_step, + ) + + def _tower_sub_parser(subparsers: "_SubParsersAction[ArgumentParser]"): subparser = subparsers.add_parser( "tower", @@ -178,6 +204,24 @@ def _towerhcraft_from_cli(args: Namespace): return env +def _treasure_sub_parser(subparsers: "_SubParsersAction[ArgumentParser]"): + subparser = subparsers.add_parser( + "treasure", + help="Treasure: A simple environment for testing.", + description="The goal of the environment is to get the gold from the chest using a key.", + ) + subparser.set_defaults(func=_treasure_from_cli) + + +def _treasure_from_cli(args: Namespace): + window = _window_from_cli(args) + env = TreasureEnv( + render_window=window, + max_step=args.max_step, + ) + return env + + def _recursive_sub_parser(subparsers: "_SubParsersAction[ArgumentParser]"): subparser = subparsers.add_parser( "recursive", diff --git a/src/hcraft/examples/__init__.py b/src/hcraft/examples/__init__.py index 51abfd75..76f612e0 100644 --- a/src/hcraft/examples/__init__.py +++ b/src/hcraft/examples/__init__.py @@ -4,13 +4,24 @@ If you built one of your own, send us a pull request so we can add it to the list! -| Gym name | CLI name | Reference | -|:----------------------------|:------------------|:----------------------------------------| -| "MineHcraft-v1" | `minecraft` |`hcraft.examples.minecraft` | -| "TowerHcraft-v1" | `tower` |`hcraft.examples.tower` | -| "RecursiveHcraft-v1" | `recursive` |`hcraft.examples.recursive` | -| "LightRecursiveHcraft-v1" | `light-recursive` |`hcraft.examples.recursive` | -| "RandomHcraft-v1" | `random` |`hcraft.examples.random_simple` | +| Gym name | CLI name | Reference | +|:-----------------------------------|:------------------|:------------------------------------------------| +| "MineHcraft-v1" | `minecraft` | `hcraft.examples.minecraft` | +| "MiniHCraftEmpty-v1" | `minicraft` | `hcraft.examples.minicraft.empty` | +| "MiniHCraftFourRooms-v1" | `minicraft` | `hcraft.examples.minicraft.fourrooms` | +| "MiniHCraftMultiRoom-v1" | `minicraft` | `hcraft.examples.minicraft.multiroom` | +| "MiniHCraftCrossing-v1" | `minicraft` | `hcraft.examples.minicraft.crossing` | +| "MiniHCraftKeyCorridor-v1" | `minicraft` | `hcraft.examples.minicraft.keycorridor` | +| "MiniHCraftDoorKey-v1" | `minicraft` | `hcraft.examples.minicraft.doorkey` | +| "MiniHCraftUnlock-v1" | `minicraft` | `hcraft.examples.minicraft.unlock` | +| "MiniHCraftUnlockPickup-v1" | `minicraft` | `hcraft.examples.minicraft.unlockpickup` | +| "MiniHCraftBlockedUnlockPickup-v1" | `minicraft` | `hcraft.examples.minicraft.unlockpickupblocked` | +| "TowerHcraft-v1" | `tower` | `hcraft.examples.tower` | +| "RecursiveHcraft-v1" | `recursive` | `hcraft.examples.recursive` | +| "LightRecursiveHcraft-v1" | `light-recursive` | `hcraft.examples.recursive` | +| "Treasure-v1" | `treasure` | `hcraft.examples.treasure` | +| "RandomHcraft-v1" | `random` | `hcraft.examples.random_simple` | + """ @@ -19,11 +30,13 @@ import hcraft.examples.random_simple as random_simple import hcraft.examples.recursive as recursive import hcraft.examples.tower as tower +import hcraft.examples.treasure as treasure from hcraft.examples.minecraft import MineHcraftEnv, MINEHCRAFT_GYM_ENVS from hcraft.examples.random_simple import RandomHcraftEnv from hcraft.examples.recursive import LightRecursiveHcraftEnv, RecursiveHcraftEnv from hcraft.examples.tower import TowerHcraftEnv +from hcraft.examples.treasure import TreasureEnv from hcraft.examples.minicraft import MINICRAFT_ENVS, MINICRAFT_GYM_ENVS EXAMPLE_ENVS = [ @@ -32,6 +45,7 @@ TowerHcraftEnv, RecursiveHcraftEnv, LightRecursiveHcraftEnv, + TreasureEnv, # RandomHcraftEnv, ] @@ -41,6 +55,7 @@ "TowerHcraft-v1", "RecursiveHcraft-v1", "LightRecursiveHcraft-v1", + "Treasure-v1", ] @@ -49,6 +64,7 @@ "minicraft", "recursive", "tower", + "treasure", "random_simple", "MineHcraftEnv", "RandomHcraftEnv", diff --git a/src/hcraft/examples/minicraft/__init__.py b/src/hcraft/examples/minicraft/__init__.py index 8e01def8..87e80448 100644 --- a/src/hcraft/examples/minicraft/__init__.py +++ b/src/hcraft/examples/minicraft/__init__.py @@ -1,6 +1,8 @@ import inspect from pathlib import Path +from typing import List, Type +from hcraft.examples.minicraft.minicraft import MiniCraftEnv from hcraft.examples.minicraft.empty import MiniHCraftEmpty from hcraft.examples.minicraft.fourrooms import MiniHCraftFourRooms from hcraft.examples.minicraft.multiroom import MiniHCraftMultiRoom @@ -12,7 +14,7 @@ from hcraft.examples.minicraft.keycorridor import MiniHCraftKeyCorridor -MINICRAFT_ENVS = [ +MINICRAFT_ENVS: List[Type[MiniCraftEnv]] = [ MiniHCraftEmpty, MiniHCraftFourRooms, MiniHCraftMultiRoom, @@ -24,6 +26,8 @@ MiniHCraftKeyCorridor, ] +MINICRAFT_NAME_TO_ENV = {env.MINICRAFT_NAME: env for env in MINICRAFT_ENVS} + __all__ = [ "MiniHCraftEmpty", "MiniHCraftFourRooms", diff --git a/src/hcraft/examples/minicraft/minicraft.py b/src/hcraft/examples/minicraft/minicraft.py index bb22a91c..61e1d1aa 100644 --- a/src/hcraft/examples/minicraft/minicraft.py +++ b/src/hcraft/examples/minicraft/minicraft.py @@ -15,6 +15,8 @@ class MiniCraftEnv(HcraftEnv): [minigrid environments](https://minigrid.farama.org/environments/minigrid/). """ + MINICRAFT_NAME = None + def __init__( self, minicraft_name: str, @@ -31,7 +33,7 @@ def __init__( If None, never truncates the episode. Defaults to None. render_window: Window using to render the environment with pygame. """ - self.minicraft_name = minicraft_name + self.MINICRAFT_NAME = minicraft_name transformations = self.build_transformations() world = world_from_transformations( transformations=transformations, @@ -40,7 +42,7 @@ def __init__( start_zones_items=start_zones_items, ) world.resources_path = os.path.join(os.path.dirname(__file__), "resources") - super().__init__(world, name=f"MiniHCraft{self.minicraft_name}", **kwargs) + super().__init__(world, name=f"MiniHCraft{self.MINICRAFT_NAME}", **kwargs) @abstractmethod def build_transformations(self) -> List[Transformation]: diff --git a/src/hcraft/examples/treasure/__init__.py b/src/hcraft/examples/treasure/__init__.py index 48e713c8..599acb29 100644 --- a/src/hcraft/examples/treasure/__init__.py +++ b/src/hcraft/examples/treasure/__init__.py @@ -7,6 +7,19 @@ from hcraft.transformation import Transformation, Use, Yield, PLAYER, CURRENT_ZONE from hcraft.world import world_from_transformations +# gym is an optional dependency +try: + import gym + + gym.register( + id="Treasure-v1", + entry_point="hcraft.examples.treasure:TreasureEnv", + ) + + +except ImportError: + pass + class TreasureEnv(HcraftEnv): """A simple environment used in for the env building tutorial."""