From df9bd2d277eb890ea3923ed74a4fddb913039dd3 Mon Sep 17 00:00:00 2001
From: Filippo Bigi <98903385+frostedoyster@users.noreply.github.com>
Date: Thu, 18 Jul 2024 12:56:08 +0200
Subject: [PATCH] Update trainer.py

---
 src/metatrain/experimental/pet/trainer.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/src/metatrain/experimental/pet/trainer.py b/src/metatrain/experimental/pet/trainer.py
index 5cf889651..b36551aa3 100644
--- a/src/metatrain/experimental/pet/trainer.py
+++ b/src/metatrain/experimental/pet/trainer.py
@@ -80,9 +80,6 @@ def train(
 
         # set model hypers
         self.hypers["ARCHITECTURAL_HYPERS"] = model.hypers
-        dtype = train_datasets[0][0]["system"].positions.dtype
-        if dtype != torch.float32:
-            raise ValueError("PET only supports float32 as dtype")
         self.hypers["ARCHITECTURAL_HYPERS"]["DTYPE"] = "float32"
 
         # set MLIP_SETTINGS