Add pusht as package

pyproject.toml

@@ -1,19 +1,23 @@
 [tool.poetry]
 name = "lerobot"
 version = "0.1.0"
-description = "Le robot is learning"
+description = "🤗 LeRobot: State-of-the-art Machine Learning for Real-World Robotics in Pytorch"
 authors = [
     "Rémi Cadène <re.cadene@gmail.com>",
+]
+maintainers = [
+    "Alexander Soare <alexander.soare159@gmail.com>",
     "Simon Alibert <alibert.sim@gmail.com>",
 ]
 repository = "https://github.com/Cadene/lerobot"
 readme = "README.md"
-license = "MIT"
+license = "Apache-2.0"
+keywords = ["robotics, deep, reinforcement, learning, pytorch"]
 classifiers=[
     "Development Status :: 3 - Alpha",
     "Intended Audience :: Developers",
     "Topic :: Software Development :: Build Tools",
-    "License :: OSI Approved :: MIT License",
+    "License :: OSI Approved :: Apache Software License",
     "Programming Language :: Python :: 3.10",
 ]
 packages = [{include = "lerobot"}]
@@ -52,11 +56,19 @@ robomimic = "0.2.0"
 gymnasium-robotics = "^1.2.4"
 gymnasium = "^0.29.1"
 cmake = "^3.29.0.1"
+sim-pusht = { version = "^0.1.0", optional = true}
+
+
+[tool.poetry.extras]
+pusht = ["sim-pusht"]
 
 
 [tool.poetry.group.dev.dependencies]
 pre-commit = "^3.6.2"
 debugpy = "^1.8.1"
+
+
+[tool.poetry.group.test.dependencies]
 pytest = "^8.1.0"
 pytest-cov = "^5.0.0"
 
@@ -93,6 +105,9 @@ exclude = [
 select = ["E4", "E7", "E9", "F", "I", "N", "B", "C4", "SIM"]
 ignore-init-module-imports = true
 
+[tool.ruff.lint.isort]
+known-first-party = ["pusht"]
+
 [tool.poetry-dynamic-versioning]
 enable = true