Fix TypedDict import for Python 3.10 compatibility

- Use typing_extensions.TypedDict instead of typing.TypedDict
- Fixes Pydantic error on Python < 3.12

atroposlib/envs/base.py

@@ -11,7 +11,8 @@ from abc import ABC, abstractmethod
 from datetime import datetime
 from enum import Enum
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Tuple, TypedDict, Union
+from typing import Any, Dict, List, Optional, Tuple, Union
+from typing_extensions import TypedDict
 
 import aiohttp
 import jsonlines