apiVersion: apps/v1
kind: Deployment
metadata:
  name: tensorboard
  namespace: bionemo-training
spec:
  replicas: 1
  selector:
    matchLabels:
      app: tensorboard
  template:
    metadata:
      labels:
        app: tensorboard
    spec:
      containers:
      - name: tensorboard
        image: tensorflow/tensorflow:latest
        command:
        - tensorboard
        args:
        - --logdir=/workspace/bionemo2/results/lightning_logs
        - --port=6006
        volumeMounts:
        - name: bionemo-storage
          mountPath: /workspace/bionemo2/results
          subPath: tensorboard-logs
          readOnly: true
      volumes:
      - name: bionemo-storage
        persistentVolumeClaim:
          claimName: bionemo-filestore
      serviceAccountName: tensorboard-sa