<reponame>mlbench/mlbench-dashboard
from django.db import models

from api.models.modelrun import ModelRun


class KubePod(models.Model):
    name = models.CharField(max_length=1000)
    labels = models.CharField(max_length=1000)
    phase = models.CharField(max_length=20)
    ip = models.CharField(max_length=15)
    node_name = models.CharField(max_length=1000)

    model_run = models.ForeignKey(
        ModelRun, related_name="pods", blank=True, null=True, on_delete=models.SET_NULL
    )

from django.db import models

from api.models.modelrun import ModelRun


class KubePod(models.Model):
    name = models.CharField(max_length=1000)
    labels = models.CharField(max_length=1000)
    phase = models.CharField(max_length=20)
    ip = models.CharField(max_length=15)
    node_name = models.CharField(max_length=1000)

    model_run = models.ForeignKey(
        ModelRun, related_name="pods", blank=True, null=True, on_delete=models.SET_NULL
    )