Remove OrderedDict from updates.py

Author: Armavica

pytensor/updates.py

@@ -1,8 +1,6 @@
 """Defines Updates object for storing a (SharedVariable, new_value) mapping."""
 
 import logging
-import warnings
-from collections import OrderedDict
 
 from pytensor.compile.sharedvalue import SharedVariable
 
@@ -12,30 +10,16 @@
 logger = logging.getLogger("pytensor.updates")
 
 
-# Must be an OrderedDict or updates will be applied in a non-deterministic
-# order.
-class OrderedUpdates(OrderedDict):
+# Relies on the fact that dict is ordered, otherwise updates will be applied
+# in a non-deterministic order.
+class OrderedUpdates(dict):
     """
     Dict-like mapping from SharedVariable keys to their new values.
 
     This mapping supports the use of the "+" operator for the union of updates.
     """
 
     def __init__(self, *key, **kwargs):
-        if (
-            len(key) >= 1
-            and isinstance(key[0], dict)
-            and len(key[0]) > 1
-            and not isinstance(key[0], OrderedDict)
-        ):
-            # Warn when using as input a non-ordered dictionary.
-            warnings.warn(
-                "Initializing an `OrderedUpdates` from a "
-                "non-ordered dictionary with 2+ elements could "
-                "make your code non-deterministic. You can use "
-                "an OrderedDict that is available at "
-                "collections.OrderedDict for python 2.6+."
-            )
         super().__init__(*key, **kwargs)
         for key in self:
             if not isinstance(key, SharedVariable):
@@ -56,19 +40,7 @@ def __setitem__(self, key, value):
     def update(self, other=None):
         if other is None:
             return
-        if (
-            isinstance(other, dict)
-            and len(other) > 1
-            and not isinstance(other, OrderedDict)
-        ):
-            # Warn about non-determinism.
-            warnings.warn(
-                "Updating an `OrderedUpdates` with a "
-                "non-ordered dictionary with 2+ elements could "
-                "make your code non-deterministic",
-                stacklevel=2,
-            )
-        for key, val in OrderedDict(other).items():
+        for key, val in dict(other).items():
             if key in self:
                 if self[key] == val:
                     continue