Use DecimalField for hyperparameter values

backend-developers-ltd · May 6, 2024 · 7ed0f44 · 7ed0f44
1 parent 09dbcd9
commit 7ed0f44
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 22 deletions.
diff --git a/app/src/bittensor_panel/core/migrations/0001_initial.py b/app/src/bittensor_panel/core/migrations/0001_initial.py
@@ -1,4 +1,4 @@
-# Generated by Django 4.2.11 on 2024-05-03 20:44
+# Generated by Django 4.2.11 on 2024-05-06 16:07
 
 from django.db import migrations, models
 
@@ -12,11 +12,19 @@ class Migration(migrations.Migration):
         migrations.CreateModel(
             name="HyperParameter",
             fields=[
-                ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
                 ("created_at", models.DateTimeField(auto_now_add=True)),
                 ("updated_at", models.DateTimeField(auto_now=True)),
                 ("name", models.CharField(max_length=255, unique=True)),
-                ("value", models.IntegerField()),
+                ("value", models.DecimalField(decimal_places=0, max_digits=32)),
             ],
         ),
     ]
diff --git a/app/src/bittensor_panel/core/migrations/0002_alter_hyperparameter_value.py b/app/src/bittensor_panel/core/migrations/0002_alter_hyperparameter_value.py
diff --git a/app/src/bittensor_panel/core/models.py b/app/src/bittensor_panel/core/models.py
@@ -5,4 +5,4 @@ class HyperParameter(models.Model):
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
     name = models.CharField(max_length=255, unique=True)
-    value = models.BigIntegerField()
+    value = models.DecimalField(max_digits=32, decimal_places=0)
diff --git a/app/src/bittensor_panel/core/services.py b/app/src/bittensor_panel/core/services.py
@@ -8,7 +8,7 @@ def update_hyperparam(instance: HyperParameter) -> None:
     Update hyperparameter in the subtensor with the new value and save changes to the database.
     """
 
-    result = update_remote_hyperparam(instance.name, instance.value)
+    result = update_remote_hyperparam(instance.name, int(instance.value))
 
     if not result:
         raise HyperParameterUpdateFailed("Failed to update remote hyperparameter. Subtensor returned False.")